【二维前缀和】小白月赛-秘法地震

引言：
矩阵前缀和
矩阵前缀和，即一个矩阵的一片区域的和，如图：

基本公式就是:pre[i][j]=pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1]+a[i][j];

题目：
秘法地震
链接：https://ac.nowcoder.com/acm/contest/2272/C
来源：牛客网

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

这种魔法可以在一片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)

思路：
求出矩阵的前缀和，然后对于每个i<=n-k+1，j<=m-k+1区间内的点(i，j)，判断它的右下方的k*k区域的前缀和是否大于1，若是的话，说明包含了建筑物。(跟暴力的思路是一样的，只是少了两层循环~~~~)
对于每个点的k*k区域求法是这样的:

int judge(int x,int y){
	int ex=x+k-1;
	int ey=y+k-1;
	return pre[ex][ey]-pre[x-1][ey]-pre[ex][y-1]+pre[x-1][y-1];
}

其实不难理解，就是将上面通式中的a[i][j]换成了一个大的k*k矩阵。
完整代码：

#include<bits/stdc++.h>
using namespace std;
template<class T>inline void read(T &res)
{
	char c;T flag=1;
	while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
	while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}

const int maxn=3005;
char b[maxn][maxn];
int a[maxn][maxn],pre[maxn][maxn];
int n,m,k,sum=0;

int judge(int x,int y){
	int ex=x+k-1;
	int ey=y+k-1;
	return pre[ex][ey]-pre[x-1][ey]-pre[ex][y-1]+pre[x-1][y-1];
}

int main(){
	read(n);
	read(m);
	read(k);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			scanf("%c",&b[i][j]);
		}
		getchar();
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			a[i][j]=b[i][j]-'0';
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			pre[i][j]=pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1]+a[i][j];
		}
	}
	for(int i=1;i<=n-k+1;i++){
		for(int j=1;j<=m-k+1;j++){
			if(judge(i,j)){
				sum++;
			}
		}
	}
	printf("%d\n",sum);
	return 0;
}