4331-Image Recognition

(http://acm.hdu.edu.cn/showproblem.php?pid=4331)

#include<iostream>  
#include<cmath>  
#include<cstdio>  
#include<vector>  
#include<cstring>  
#include<algorithm>  
using namespace std;  

int t,T,ans,s[1005][1005],n;  
int d1[1005][1005],d2[1005][1005],du[1005][1005],dd[1005][1005],dl[1005][1005],dr[1005][1005];  
//du(up) dd(down) dl(left) dr(right)表示该点向四个方向最长有几个1  
//d1表示该点向右下对角线的最长边长(即该点向下和向右短的那条边) d2表示向左上  
int main()  
{  
    int i,j,k;  
    scanf("%d",&T);  
    t=0;  
    while(T--)  
    {  
        ans=0;  
        t++;  
        scanf("%d",&n);  
        for(i=0;i<n;i++)  
            for(j=0;j<n;j++)  
            {  
                scanf("%d",&s[i][j]);  
                if(s[i][j]) ans++;  
            }  
        memset(du,0,sizeof du);  
        memset(dd,0,sizeof dd);  
        memset(dr,0,sizeof dr);  
        memset(dl,0,sizeof dl);  
        for(i=0;i<n;i++)//右下方向  
        {  
            for(j=0;j<n;j++)  
            {  
                if(s[i][j])  
                {  
                    dd[i][j]=dd[i-1][j]+1;  
                    dr[i][j]=dr[i][j-1]+1;  
                }  
                d1[i][j]=min(dd[i][j],dr[i][j]); //得到可能的最长边长  
            }  
        }  
        for(i=n-1;i>=0;i--)//左上方向  
        {  
            for(j=n-1;j>=0;j--)  
            {  
                if(s[i][j])  
                {  
                    du[i][j]=du[i+1][j]+1;  
                    dl[i][j]=dl[i][j+1]+1;  
                }  
                d2[i][j]=min(du[i][j],dl[i][j]); //得到可能的最长边长 
            }  
        }  
        for(i=0;i<n;i++)  
        {  
            for(j=0;j<n;j++)  
            {  
                for(k=2;k<=d2[i][j];k++)//枚举边长，同时也枚举了对角线  
                {  
                    if(k<=d1[i+k-1][j+k-1])  
                        ans++;  
                }  
            }  
        }  
        printf("Case %d: %d\n",t,ans);  
    }  
    return 0;  
}