HDU 4331 Contest 4

一个很直观的想法是，求出每个点上下左右能到达的最大长度。然后枚举其斜边。。。没想到过了。。。。

当然，题解有一个很巧妙的优化，利用树状数组，那个太巧妙了。

#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],dt[1005][1005],dd[1005][1005],dl[1005][1005],dr[1005][1005];
//dt(top) 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(dt,0,sizeof dt);
        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])
                {
                    dt[i][j]=dt[i+1][j]+1;
                    dl[i][j]=dl[i][j+1]+1;
                }
                d2[i][j]=min(dt[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;
}