[三个数组的动归］农田个数

[描述］

你的老家在河北农村。过年时，你回老家去拜年。你家有一片N*M农田，将其看成一个N*M的方格矩阵，有些方格是一片水域。你的农村伯伯听说你是学计算机的，给你出了一道题： 他问你：这片农田总共包含了多少个不存在水域的正方形农田。两个正方形农田不同必须至少包含下面的两个条件中的一条：1.边长不相等2.左上角的方格不是同一方格

[输入格式］

输入数据第一行为两个由空格分开的正整数N、M（1<=m<n<=1000）第2行到第N+1行每行有M个数字（0或1），描述了这一片农田。0表示这个方格为水域，否则为农田（注意：数字之间没有空格，而且每行不会出现空格）

[输出格式］

满足条件的正方形农田个数。

[样例输入］

3 3

110

110

000

[样例输出］

5

[分析］

动态规划。f[i][j]表示以i,j为右下角能构成的最大的正方形的边长。r[i][j]表示从i,j向左能有多长都是1。d[i][j]表示从i,j向上能扩展多少个都是1。

#include <stdio.h>

#define maxn 1010



bool a[maxn][maxn];

char c;

int n,m;

int f[maxn][maxn],r[maxn][maxn],d[maxn][maxn];

int ans;



int main()

{

    freopen("count.in","r",stdin);

    freopen("count.out","w",stdout);

    

    scanf("%d%d",&n,&m);

    scanf("%c",&c);

    for (int i=1;i<=n;++i)

    {

        for (int j=1;j<=m;++j)

        {

            scanf("%c",&c);

            if (c=='1') a[i][j]=1;

            else a[i][j]=0;

        }

        scanf("%c",&c);

    }

    

    for (int i=1;i<=n;++i)

        for (int j=1;j<=m;++j)

            if (a[i][j])

            {

                f[i][j]=f[i-1][j-1]+1;

                r[i][j]=r[i][j-1]+1;

                d[i][j]=d[i-1][j]+1;

                if (r[i][j]<f[i][j]) f[i][j]=r[i][j];

                if (d[i][j]<f[i][j]) f[i][j]=d[i][j];

            }

    

    for (int i=1;i<=n;++i)

        for (int j=1;j<=m;++j)

            ans+=f[i][j];

    printf("%d\n",ans);

    

    return 0;

}