数组题目：图像重叠

题目

标题和出处

标题：图像重叠

出处：835. 图像重叠

难度

6 级

题目描述

要求

给出两个图像 $\text{img1}$ 和 $\text{img2}$，都是大小为 $\text{n}\times \text{n}$ 的正方形二进制矩阵（二进制矩阵只包含 $\text{0}$ 和 $\text{1}$）。

我们转换其中一个图像，向左、右、上、下移动任何数量的单位，并把它放在另一个图像的上面。转换之后，该转换的重叠是指两个图像中都为 $\text{1}$ 的位置的数目。

（请注意，转换不包括任何方向的旋转。）

最大可能的重叠是多少？

示例

示例 1：

输入：$\text{img1 = [[1,1,0],[0,1,0],[0,1,0]], img2 = [[0,0,0],[0,1,1],[0,0,1]]}$
输出：$\text{3}$
解释：我们将 $\text{img1}$ 向右移动 $\text{1}$ 个单位，向下移动 $\text{1}$ 个单位。

两个图像中都为 $\text{1}$ 的位置的数目是 $\text{3}$（红色的位置）。

示例 2：

输入：$\text{img1 = [[1]], img2 = [[1]]}$
输出：$\text{1}$

示例 3：

输入：$\text{img1 = [[0]], img2 = [[0]]}$
输出：$\text{0}$

数据范围

• $\text{n}=\text{img1.length}$
• $\text{n}=\text{img1[i].length}$
• $\text{n}=\text{img2.length}$
• $\text{n}=\text{img2[i].length}$
• $\text{1}\le \text{n}\le \text{30}$
• $\text{img1[i][j]}$ 是 $\text{0}$ 或 $\text{1}$
• $\text{img2[i][j]}$ 是 $\text{0}$ 或 $\text{1}$

解法

思路和算法

将一个图像进行转换，等价于将另一个图像以相反方向进行转换，因此只需要考虑转换 ${\text{img}}_1$ 的情况。

在转换 ${\text{img}}_1$ 之后，只有当 ${\text{img}}_1$ 和 ${\text{img}}_2$ 有至少一个元素重合的时候，重叠才可能大于 $0$。由于 ${\text{img}}_1$ 和 ${\text{img}}_2$ 都是 $n$ 行 $n$ 列的正方形矩阵，为了在转换之后使得两个图像有重合，转换时在水平方向和竖直方向可以移动的最大单位数都必须小于 $n$，即不能超过 $n-1$。

用 $(i,j)$ 表示 ${\text{img}}_1$ 的转换，其中 $-(n-1)\le i,j\le n-1$。

• 竖直方向，$i<0$ 表示 ${\text{img}}_1$ 向上移动，$i>0$ 表示 ${\text{img}}_1$ 向下移动；

• 水平方向，$j<0$ 表示 ${\text{img}}_1$ 向左移动，$j>0$ 表示 ${\text{img}}_1$ 向右移动。

转换之后，${\text{img}}_1[\text{row}-i][\text{column}-j]$ 和 ${\text{img}}_2[\text{row}][\text{column}]$ 重合，如果这两个值都是 $1$，则为一个重叠。

由于行下标和列下标都要满足大于等于 $0$ 且小于 $n$，因此需要满足以下条件：

• $0\le \text{row}-i<n$；

• $0\le \text{column}-j<n$；

• $0\le \text{row}<n$；

• $0\le \text{column}<n$。

因此枚举行下标和列下标时应满足以下条件：

• $\max (0,i)\le \text{row}<\min (n,n+i)$；

• $\max (0,j)\le \text{column}<\min (n,n+j)$。

对于每组 $(i,j)$，确定 $\text{row}$ 和 $\text{column}$ 的范围之后，即可遍历范围内的全部 $(\text{row},\text{column})$，计算该组 $(i,j)$ 对应的转换的重叠。

遍历全部 $(i,j)$ 之后即可得到最大可能的重叠。

代码

class Solution {
    public int largestOverlap(int[][] img1, int[][] img2) {
        int maxOverlap = 0;
        int n = img1.length;
        int maxMove = n - 1;
        for (int i = -maxMove; i <= maxMove; i++) {
            for (int j = -maxMove; j <= maxMove; j++) {
                int overlap = 0;
                int rowStart = Math.max(0, i), rowEnd = Math.min(n, n + i);
                int columnStart = Math.max(0, j), columnEnd = Math.min(n, n + j);
                for (int row = rowStart; row < rowEnd; row++) {
                    for (int column = columnStart; column < columnEnd; column++) {
                        if (img1[row - i][column - j] == 1 && img2[row][column] == 1) {
                            overlap++;
                        }
                    }
                }
                maxOverlap = Math.max(maxOverlap, overlap);
            }
        }
        return maxOverlap;
    }
}

复杂度分析

• 时间复杂度：$O(n^4)$，其中 $n$ 是矩阵 ${\text{img}}_1$ 和 ${\text{img}}_2$ 的边长。枚举水平方向和竖直方向的移动单位数各需要 $O(n)$ 的时间，因此所有可能的转换数为 $O(n^2)$，对于每种转换，需要 $O(n^2)$ 的时间计算两个图像的重叠，因此总时间复杂度是 $O(n^4)$。

• 空间复杂度：$O(1)$。