[描述]
你的老家在河北农村。过年时,你回老家去拜年。你家有一片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;
}