直接二维前缀和,判断区间是不是>0即可.
#include <bits/stdc++.h>
using namespace std;
const int N=1e3+50;
char s[N][N];
int w[N][N];
int f[N][N];
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
cin>>s[i][j];
if(s[i][j]=='1') w[i][j]=1;
else w[i][j]=0;
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
f[i][j]=f[i-1][j]+f[i][j-1]-f[i-1][j-1]+w[i][j];
}
}int ans=0;
for(int i=k;i<=n;i++)
{
for(int j=k;j<=m;j++)
{
if(f[i][j]-f[i-k][j]-f[i][j-k]+f[i-k][j-k]) ans++;
}
}printf("%d\n",ans);
return 0;
}
京公网安备 11010502036488号