题目描述:
给出两个图像 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;
}
};