图像重叠

给出两个图像 A 和 B ,A 和 B 为大小相同的二维正方形矩阵。(并且为二进制矩阵,只包含0和1)。

我们转换其中一个图像,向左,右,上,或下滑动任何数量的单位,并把它放在另一个图像的上面。之后,该转换的重叠是指两个图像都具有 1 的位置的数目。
(请注意,转换不包括向任何方向旋转。)
最大可能的重叠是什么?

示例 1:
输入:A = [[1,1,0],
[0,1,0],
[0,1,0]]
B = [[0,0,0],
[0,1,1],
[0,0,1]]
输出:3
解释: 将 A 向右移动一个单位,然后向下移动一个单位。
注意:
1 <= A.length = A[0].length = B.length = B[0].length <= 30
0 <= A[i][j], B[i][j] <= 1

解题思路是,计算出所有从A中1到B中1的平移向量,那么出现次数最多的平移向量的频率,就是可重叠1的最大个数

class Solution:
    def largestOverlap(self, A: List[List[int]], B: List[List[int]]) -> int:
        d=collections.defaultdict(int)
        a=[]
        b=[]
        for i in range(len(A)):
            for j in range(len(A[0])):
                if A[i][j]==1:
                    a.append((i,j))
                if B[i][j]==1:
                    b.append((i,j))
        ans=0
        for x in a:
            for y in b:
                t=(y[0]-x[0],y[1]-x[1])
                d[t]+=1
                ans=max(ans,d[t])
        return ans
©️2020 CSDN 皮肤主题: 游动-白 设计师:上身试试 返回首页