计算机学院大学生程序设计竞赛（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=+;
char s[maxn][maxn];
int n,m;
int c[maxn][maxn];
int sum[*maxn][*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,,sizeof sum);
        memset(ans,,sizeof ans);
    //    memset(c,0,sizeof c);
        for(int i=;i<n;i++) scanf("%s",s[i]);
        for(int i=;i<=n;i++) for(int j=;j<=n;j++) c[i][j]=s[i-][j-]-'';

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

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

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

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

    }
    return ;
}