牛客小白月赛19 弱版 C

  https://ac.nowcoder.com/acm/contest/2272/C

题意：：

帕秋莉掌握了一种土属性魔法 

这种魔法可以在一片k×k大小的一个正方形区域内产生地震

但是如果某片即将产生地震的区域内有建筑物，帕秋莉会停止施法

整个地图大小为n×m，其中一些地方有建筑

请问有多少种可能的情况，使得帕秋莉会停止施法

思路：：二维前缀和思想 ，记录在任意的 K*K 的区间内是否存在1

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e3+5;

char a[maxn];
int num[maxn][maxn];
int b[maxn][maxn];
int n,m,k;
int check(int x,int y)
{
    return num[x+k-1][y+k-1]-num[x-1][y+k-1]-num[x+k-1][y-1]+num[x-1][y-1];
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++){
        scanf("%s",a+1);
        for(int j=1,p=strlen(a+1);j<=p;j++){
            b[i][j]=a[j]-'0';
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            num[i][j]=b[i][j]+num[i-1][j]+num[i][j-1]-num[i-1][j-1];
        }
    }
    int ans=0;
    for(int i=1;i+k-1<=n;i++){
        for(int j=1;j+k-1<=m;j++){
            if(check(i,j)){ans++;}
        }
    }
    cout<<ans<<endl;
    return 0;
}