洛谷P1387 最大正方形

//暴力 二维前缀和+枚举 O(N^3)
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=105;
int n,m,ans,a[maxn][maxn],sum[maxn][maxn];
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i) for(int j=1;j<=m;++j){
		scanf("%d",&a[i][j]);
		sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
	}
	int tmp=min(n,m);
	for(int i=1;i<=n;++i) for(int j=1;j<=m;++j)for(int len=1;len<=tmp;++len){
		int x=i+len-1,y=j+len-1;
		if(x>n||y>m||sum[x][y]-sum[i-1][y]-sum[x][j-1]+sum[i-1][j-1]!=len*len)break;
		ans=max(ans,len);
	}
	printf("%d",ans);
	return 0;
}