题目描述
帕秋莉掌握了一种土属性魔法
这种魔法可以在一片k×k大小的一个正方形区域内产生地震
但是如果某片即将产生地震的区域内有建筑物,帕秋莉会停止施法
整个地图大小为n×m,其中一些地方有建筑
请问有多少种可能的情况,使得帕秋莉会停止施法
输入描述:
第一行三个数n, m, k,意义见描述
接下来一个n×m的01矩阵表示这篇区域的情况,1表示这个地方有建筑
输出描述:
输出一个数表示答案
输入
4 4 2
1000
0100
0000
0001
输出
5
思路:算法核心:前缀和
本质最大子矩阵求大于0 的矩阵数量,详细解法见激光炸弹
直接上代码(二次元题目背景真不错,请加大力度 ,含注释)
//土 秘法地震
//算法:前缀和
//本质:最大子矩阵,再本质:最大连续子串
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int a[1005][1005];//矩阵
int main()
{
int i,j,k,n,m,cnt=0;//cnt计数符合情况的数量
string str;
cin>>n>>m>>k;
for(i=1;i<=n;i++)
{
cin>>str;
for(j=0;j<m;j++)//将输入数据转化为行列为11开头的矩阵
{
a[i][j+1]=a[i][j]+str[j]-'0';//转化过程中,直接计算行前缀和
}
}
for(j=1;j<=m;j++)
for(i=2;i<=n;i++)
a[i][j]+=a[i-1][j];//计算列前缀和
for(i=0;i<=n-k;i++)//检查符合的情况
for(j=0;j<=m-k;j++)
if(a[i+k][j+k]+a[i][j]-a[i+k][j]-a[i][j+k]>0) cnt++;//符合情况
cout<<cnt<<endl;
}