思路
对于本题,暴力的想法是遍历每个邮票大小的区域,如果这个区域都是空,那就给这个区域的每个位置贴上邮票,最后如果有没有贴邮票且为空的地方,就是false;
但数据范围这么大,暴力的操作必然超时,对于这样的批量操作,用前缀和和差分来记录批量重复的操作;
首先用二位前缀和记录每个位置对应的前缀和,记录是否已被占据,如果邮票大小范围内二维前缀和没有变化,
即sum[i][j] - sum[i-Height][j] - sum[i][j-Width] + sum[i-Height][j-Width] == 0
说明原数组grid中这个区域没有1,即这个范围没有被占据;
对于有空位的区域,用另一个数组diff(也就是差分)标记这个区域都贴上邮票了,具体操作为左上角和右下角+1,左下角和右上角-1,这样求出的前缀和就能表示这个区域内都进行过标记;
求出差分数组的前缀和,如果对应位置没有邮票进行过标记,且原本位置是空的,说明没有贴满,否则就是全贴满了。
代码
class Solution {
public:
bool possibleToStamp(vector<vector<int>>& grid, int Height, int Width) {
int m = grid.size(), n = grid[0].size();
vector<vector<int>> sum(m+1, vector<int>(n+1, 0));
vector<vector<int>> diff(m+2, vector<int>(n+2, 0));
for(int i = 1; i <= m; i++) {
for(int j = 1; j <= n; j++) {
sum[i][j] = sum[i][j-1] + grid[i-1][j-1];
}
}
for(int j = 1; j <= n; j++) {
for(int i = 1; i <= m; i++) {
sum[i][j] += sum[i-1][j];
}
}
for(int i = Height; i <= m; i++) {
for(int j = Width; j <= n; j++) {
int cnt = sum[i][j] - sum[i-Height][j] - sum[i][j-Width] + sum[i-Height][j-Width];
//这个区间内都是0,可以贴邮票
if(cnt == 0) {
diff[i+1][j+1] += 1;
diff[i-Height+1][j+1] -= 1;
diff[i+1][j-Width+1] -= 1;
diff[i-Height+1][j-Width+1] += 1;
}
}
}
for(int i = 1; i <= m; i++) {
for(int j = 1; j <= n; j++) {
diff[i][j] += diff[i][j-1];
}
}
for(int j = 1; j <= n; j++) {
for(int i = 1; i <= m; i++) {
diff[i][j] += diff[i-1][j];
if(grid[i-1][j-1] == 0 && diff[i][j] == 0) return false;
}
}
return true;
}
};