ssl1633农田个数

题目大意

给你一个由0和1组成的n*m的矩阵，求其中以1组成的正方形个数。
其中不同的两个正方形农田至少要包含下面的两个条件中的一条：
边长不相等;
左上角的方格不是同一方格

解题思路

如果直接枚举求个数肯定会爆，所以我们要找其他的做法，经过观察和实践，不难发现，对于一个右下角为$(i,j)$的非0正方形，它的最大边长为$min(f_{i-1,j},min(f_{i{,}_j-1},f_{i-1,j-1}))+1$，若它是0，则它无法构成正方形。最后只需把所有边长加起来即可，因为当前的最大边长若是$n$，则边长为$n-1,n-2,\dots ,1$的正方形都一定会出现。

代码

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int n,m,a[1005][1005],f[1005][1005],ans;
char c;
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			cin>>c;
			a[i][j]=c-'0';
		}
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			if(!a[i][j]) f[i][j]=0;//若右下角是0则不能构成正方形
			else f[i][j]=min(f[i-1][j],min(f[i-1][j-1],f[i][j-1]))+1;//最大周长
			ans+=f[i][j];//累加
		}
	cout<<ans;
	return 0;
}