N-Puzzle问题中3D数组逐单元格匹配计数的高效实现求助

对于你的需求，直接用嵌套zip对齐逐位比较配合sum累加的写法就可以，时间复杂度为O(n*m)（n、m分别为网格的行、列数），属于该场景下的最优复杂度，同时没有额外空间开销，效率很高。

代码示例 #

goal = [0, 2, [[3, 2, 0], [6, 1, 8], [4, 7, 5]]]
test = [0, 0, [[0, 7, 1], [4, 3, 2], [8, 6, 5]]]

match_count = sum(
    g_val == t_val
    for g_row, t_row in zip(goal[2], test[2])
    for g_val, t_val in zip(g_row, t_row)
)

print(match_count) # 输出结果为1，和你示例的预期一致

补充说明 #

如果你的需求里需要排除空白格0的匹配，只需要在比较逻辑里加一层过滤即可：

match_count = sum(
    g_val == t_val and g_val != 0
    for g_row, t_row in zip(goal[2], test[2])
    for g_val, t_val in zip(g_row, t_row)
)

内容的提问来源于stack exchange，提问作者user13413898