如何实现高效的 Python 算法来查找匹配数字？

Question

假设我有 2 个相同数字 A 和 B（两个数字不同），我需要找到 X（A 和 B 中的位数，并且位置相同）和 Y（ A 和 B 中的数字，但位置不同）。 例如，如果 A = 4567 且 B = 4567，则 X 为 4，Y 为 0。如果 A = 3456 且 B = 4567，则 X 为 0，Y 为 3。不考虑效率，我的代码如下如下：

def compare(num1, num2):
  x = 0
  y = 0
  for i in range(4):
     if str(num1)[i] in str(num2) and str(num1)[i] == str(num2)[i]:
        x += 1
     elif str(num1)[i] in str(num2):
        y += 1
  return ('X = %d, Y = %d'%(x,y))

但是，我认为这不是解决此问题的最有效方法。谁能提供更有效的解决方案？谢谢...

Answer 1

我认为只有一种方法可以在保持整洁的同时缩短它。

for i in range(4):
    if str(num1)[i] == str(num2)[i]:
        x += 1
    elif str(num1)[i] in str(num2):
        y += 1

编辑在第 2 行，在 for 循环中。你写了if str(num1)[i] in str(num2) and str(num1)[i] == str(num2)[i]:。 str(num1)[i] in str(num2) 不需要，因为如果 num1[i] 等于 num2[i]，则表示 num1[i] 已经在 num2 中，所以这行代码没用。

如果你只是想缩短线路，你可以试试这个：

for i in range(4):
        if str(num1)[i] == str(num2)[i]: x += 1
        elif str(num1)[i] in str(num2): y += 1

和

x = y = 0

Answer 2

使用dict 可以减少比较次数。

str(num1)[i] in str(num2): 再次扫描整个字符串。你可以避免这种情况。

def compare(num1, num2):
    x = 0
    y = 0
    result = {}

    num1 = str(num1)
    num2 = str(num2)

    for i in range(0,len(num1)):
        result[num1[i]] = i

    for i in range(0,len(num2)):
        if num2[i] in result:
            if result[num2[i]] == i:
                x +=1;
            else:
                y +=1;

    return ('X = %d, Y = %d'%(x,y))


print(compare(4567,4567))
print(compare(3456,4567))