leetcode 835. 图像重叠(Image Overlap)

目录

• 题目描述：
• 示例 1:
• 解法：

题目描述：

给出两个图像 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

---

解法：

class Solution {
public:
    int overlap(vector<vector<int>>& A, vector<vector<int>>& B, int xa, int ya){
        int m = A.size();
        int n = A[0].size();
        int res = 0;
        for(int i = 0; i < m; i++){
            if(xa + i >= m){
                break;
            }
            for(int j = 0; j < n; j++){
                if(ya + j >= n){
                    break;
                }
                if(B[i][j] == 1 && A[xa + i][ya + j] == B[i][j]){
                    res++;
                }
            }
        }
        return res;
    }
    
    int largestOverlap(vector<vector<int>>& A, vector<vector<int>>& B) {
        int m = A.size();
        int n = A[0].size();
        int res = 0;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                res = max(res, overlap(A, B, i, j));
                res = max(res, overlap(B, A, i, j));
            }
        }
        return res;
    }
};