【SSL】1633农田个数

【SSL】1633农田个数

Time Limit:1000MS

Memory Limit:65536K

Description

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

Input

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

Output

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

Sample Input

3 3                                               
110
110
000 

Sample Output

5

思路

(1)设f[i][j]表示以方格（i,i）为右下角，可以得到的最大无水正方形边长，那么显然如果（i,j）是水域，f[i][j]=0；否则f[i][j]=min(f[i-1][j]，f[i][j-1]，f[i-1][j-1])+1
(2)求出了f数组的值后，我们可以用f1[i]表示f数组中，值为I的个数。
(3)显然，假定边长为I的正方形数目为Sum[i]，那么有Sum[i]=Sum[i+1]+F1[i]
(4)最后只要算出Sum数组各个值的和为问题的解。

代码

#include<cstdio>
#include<cstring> 
#include<cstdlib>
#include<iostream>
using namespace std;
long long n,m,f[1010][1010],f1[1010],sum[1010],ans=0;
char a[1010][1010];
void input()
{
	long long i,j;
	memset(f,0,sizeof(f));
	memset(f1,0,sizeof(f1));
	memset(sum,0,sizeof(sum));
	scanf("%lld%lld",&n,&m);
	for(i=1;i<=n;i++)
	{
		getchar();
		for(j=1;j<=m;j++)
			a[i][j]=getchar();
	}
	return;
}
void DP()
{
	int i,j;
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
		{
			if (a[i][j]=='1')
				f[i][j]=min(f[i-1][j-1],min(f[i-1][j],f[i][j-1]))+1;//状态转移方程
			else
				f[i][j]=0;
			f1[f[i][j]]++;
		}
	for(i=n;i>0;i--)//求和
	{
		sum[i]=sum[i+1]+f1[i];
		ans+=sum[i];
	}
	return;
}
int main()
{
	input();
	DP();
	printf("%lld",ans);
	return 0;
}