牛客竞赛---[土]法秘震

题目描述

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

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

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

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

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

输入描述:

第一行三个数n, m, k，意义见描述
接下来一个n×m的01矩阵表示这篇区域的情况，1表示这个地方有建筑

输出描述:

输出一个数表示答案

示例1

输入

复制4 4 2 1000 0100 0000 0001

4 4 2
1000
0100
0000
0001

输出

复制5

5

备注:

对于30%的数据，n, m≤30
对于100%的数据，n, m≤1000，k≤min(n, m)

题解：本题就是在指定的二维区间内出现了大于0的值就是所要求的，对于该题其实也就是二维前缀的考量。

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int a[1000][1000];
int sum[1010][1010];//前缀和

int main()
{
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			scanf("%1d",&a[i][j]);
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			sum[i][j]=sum[i-1][j]+sum[i][j-1]+a[i][j]-sum[i-1][j-1];
		}
	}
//	cout<<"sum"<<endl;
//	for(int i=1;i<=n;i++){
//		for(int j=1;j<=m;j++){
//			cout<<sum[i][j]<<" ";
//		}
//		cout<<endl;
//	}
	int su=0;
	for(int i=1;i<=n-k+1;i++){
		for(int j=1;j<=m-k+1;j++){
			int x1=i,y1=j,x2=i+k-1,y2=j+k-1;
			int ans=sum[x2][y2]+sum[x1-1][y1-1]-sum[x1-1][y2]-sum[x2][y1-1];
			if(ans>0){
				su++;
			}
		}
	}
	cout<<su;
	return 0;
}