传送门
我也不知道它为什么是dp。。。显然01矩阵中全为1的正方形有一个性质,那就是权值和为边长平方。那乱搞一下不就行了吗。。。
注意:ans初值赋为1。
优化:对边长(k)的枚举可以改为二分查找,因为满足条件的正方形的存在性是单调的
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=102;
int n,m,a[MAXN][MAXN],sum[MAXN][MAXN];
inline int read() {
int x=0;char c=getchar();
while (c<'0'||c>'9') c=getchar();
while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x;
}
int main() {
// freopen("P1387.in","r",stdin);
memset(sum,0,sizeof(sum)),
n=read(),m=read();
for (int i=1;i<=n;++i)
for (int j=1;j<=m;++j)
a[i][j]=read(),sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
int ans=1;
for (int i=1;i<=n;++i)
for (int j=1;j<=m;++j) {
int len=min(n-i,m-j);
for (int k=1;k<=len;++k)
if (sum[i+k][j+k]-sum[i-1][j+k]-sum[i+k][j-1]+sum[i-1][j-1]==(k+1)*(k+1)) ans=max(ans,k+1);
}
printf("%d\n",ans);
return 0;
}
还有个O(n^2)的二维dp,dp[i][j]表示以(i,j)为右下顶点的可行正方形最大边长。
#include<bits/stdc++.h>
using namespace std;
const int N=104;
int n,m,dp[N][N],a[N][N],ret=0;
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]);
for (int i=1;i<=n;++i)
for (int j=1;j<=m;++j)
dp[i][j]=a[i][j]?min(min(dp[i][j-1],dp[i-1][j]),dp[i-1][j-1])+1:0,ret=max(ret,dp[i][j]);
cout<<ret<<endl;
return 0;
}