「土」秘法地震----二维前缀和

第二天叫醒我的不是闹钟，是梦想！

题目描述
帕秋莉掌握了一种土属性魔法

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

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

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

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

输入描述:
第一行三个数n, m, k，意义见描述
接下来一个n×m的01矩阵表示这篇区域的情况，1表示这个地方有建筑
输出描述:
输出一个数表示答案
示例1
输入
复制
4 4 2
1000
0100
0000
0001
输出
复制
5
备注:
对于30%的数据，n, m≤30
对于100%的数据，n, m≤1000，k≤min(n, m)

#include<bits/stdc++.h>
using namespace std;
const int N=1100;
int a[N][N];
char s[N][N];
int n,m,k;
int check(int x1,int y1,int x2,int y2)
{
  return a[x2][y2]-a[x2][y1-1]-a[x1-1][y2]+a[x1-1][y1-1];
}
int main()
{
    char c;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++) cin>>s[i][j];
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)
        {
          a[i][j]=s[i][j]-'0';
          a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
        }
    int res=0;
    for(int i=1;i+k-1<=n;i++)
      for(int j=1;j+k-1<=m;j++)
      {

        if(check(i,j,i+k-1,j+k-1)) res++;
      }
      cout<<res<<endl;
}