题意:给定一个固定大小的h*w矩形,以及一个由01组成的矩阵,问能否用该矩形填满矩阵中的所有0;
要求:
- 矩形可以重叠,但不能覆盖1
- 不能超过矩阵的边界
- 矩形不能旋转
输入:grid = [[1,0,0,0],[1,0,0,0],[1,0,0,0],[1,0,0,0],[1,0,0,0]], stampHeight = 4, stampWidth = 3
输出:true
解释:
1 0 0 0
1 0 0 0
1 0 0 0
1 0 0 0
1 0 0 0
我们放入两个有重叠部分的邮票,能覆盖所有与空格子。
题解:二维前缀和+二维差分
通过暴力的方法,遍历这个矩阵,只要发现能够放置矩形,就放进去,遍历完成后只要矩阵中所有的0都覆盖了矩形就说明能够覆盖,所以问题转化为:判断能否覆盖、记录覆盖的情况;
- 通过二位前缀和,判断以i、j为左上角的矩形内的sum值是否为0
- 通过二维差分,记录放置矩形的情况,左上、右下加一,左下、右上减一
bool possibleToStamp(vector<vector<int>>& grid, int h, int w) {
int n = grid.size(), m = grid[0].size();
vector<vector<int>> sums(n+2, vector<int>(m+2)), dif(sums);
for(int i = 0; i < n; ++i)
for(int j = 0; j < m; ++j)
sums[i+1][j+1] = sums[i][j+1]+sums[i+1][j]-sums[i][j]+grid[i][j];
for(int i = h; i <= n; ++i) {
for(int j = w; j <= m; ++j) {
int cnt = sums[i][j]-sums[i-h][j]-sums[i][j-w]+sums[i-h][j-w];
if(cnt) continue;
dif[i-h+1][j-w+1] += 1;
dif[i-h+1][j+1] -= 1;
dif[i+1][j-w+1] -= 1;
dif[i+1][j+1] += 1;
}
}
for(int i = 1; i <= n; ++i) {
for(int j = 1; j <= m; ++j) {
dif[i][j] += dif[i][j-1]-dif[i-1][j-1]+dif[i-1][j];
if (!grid[i-1][j-1] && !dif[i][j])
return false;
}
}
return true;
}