牛客每日一题 5月18日「土」秘法地震（二维前缀和）_土体8757452416477 8555 65445555557544254546545555555-CSDN博客

「土」秘法地震

题目

在这里插入图片描述

二维前缀和

f[i][j]为前i行前j列的格子里面数字的和，f[i][j] = f[i-1][j] + f[i][j-1] - f[i-1][j-1]f[i][j]=f[i−1][j]+f[i][j−1]−f[i−1][j−1]
如图一：
在这里插入图片描述

如果我们需要求一个左上角为(a,b)右下角为(c, d)的子矩形的和，显然它等于f[c][d] - f[a][b-1]-f[a-1][b]+f[i][j]f[c][d]−f[a][b−1]−f[a−1][b]+f[i][j]
如图二：
在这里插入图片描述

题解

看每个正方形是不是有1，我们可以进行预处理二维前缀和。查询每个正方形是不是有1，只需要看正方形内的和是不是大于0，如果大于0， ans++，最后输出ans。时间复杂度为O(1)。

代码

#include <cstdio>
#include <iostream>
using namespace std;

char a[1005][1005];	
int s[1005][1005];
int n, m, k;

int main() {
    cin >> n >> m >> k;
    for(int i = 1; i <= n; i++) {
            cin >> a[i]+1;
    }
    for(int i = 1; i <= n; i++) {
        for(int j = 1; j <= m; j++) {
                s[i][j] = s[i][j-1] + s[i-1][j] - s[i-1][j-1];	//预处理二维前缀和
                if(a[i][j] == '1')
                    s[i][j]++;
        }
    }
    int ans = 0;
    for(int i = 1; i + k - 1 <= n; i++) {
        for(int j = 1; j + k - 1 <= m; j++) {
            int x = i + k - 1;
            int y = j + k - 1;
            if(s[x][y] - s[i-1][y] - s[x][j-1] + s[i-1][j-1])	// 查询k*k的正方形内是否有1
                ans++;	//如果有就ans++
        }
    }
    cout << ans << endl;
    return 0;
}