计算机学院大学生程序设计竞赛（2015’12） 1006 01 Matrix

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<map>
#include<queue>
using namespace std;

const int maxn=1000+10;
char s[maxn][maxn];
int n,m;
int c[maxn][maxn];
int sum[2*maxn][2*maxn];
int ans[maxn];

int main()
{
int T;
//freopen("F:\\in.txt","r",stdin);
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
memset(sum,0,sizeof sum);
memset(ans,0,sizeof ans);
//    memset(c,0,sizeof c);
for(int i=0;i<n;i++) scanf("%s",s[i]);
for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) c[i][j]=s[i-1][j-1]-'0';

for(int i=1;i<=2*n;i++)
for(int j=1;j<=2*n;j++)
{
if(i>n||j>n)sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
else sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+c[i][j];
}

for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(c[i][j]==0) continue;
int left,right,mid;
int anslen;
left=1;right=n;
while(left<=right)
{
mid=(left+right)/2;
int s=sum[i+mid-1][j+mid-1]-sum[i+mid-1][j-1]-sum[i-1][j+mid-1]+sum[i-1][j-1];
if(s==mid*mid)
{
anslen=mid;
left=mid+1;
}
else right=mid-1;
}
ans[anslen]++;
}
}

for(int i=n-1;i>=1;i--) ans[i]=ans[i]+ans[i+1];
//for(int i=1;i<=n;i++) printf("%d %d\n",i,ans[i]);

for(int i=1;i<=m;i++)
{
int x;
scanf("%d",&x);
printf("%d\n",ans[x]);
}

}
return 0;
}