QAQ
先用DP方法处理出矩阵的前缀和
我们现在想一个问题
既然矩阵里只能有1
那么的话判断一个矩阵是不是我们要找的正方形
我们只需要知道整个矩阵的前缀和,如果前缀和为长度的平方,那么这个正方形就是符合要求的正方形
还有一个优化,你发现组不成正方形的时候,直接停止枚举(小的都不成功,大的肯定不成啊!!1)
#include <cstdio>
#include <iostream>
using namespace std;
int map[1000][1000];
int s[1000][1000];
void screen(int n,int m)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+map[i][j];
}
int main()
{
int n,m;
int ans=1;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&map[i][j]);
screen(n,m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
for(int k=ans;i+k<=n&&j+k<=m;k++)
if((s[i+k][j+k]+s[i][j]-s[i+k][j]-s[i][j+k])==(k*k)) ans=max(ans,k);
else break;
printf("%d",ans);
return 0;
}
方法二:使用DP来搞一波
令 f[i][j]为以(i,j)的点为右下角的正方形的最大长度
如果map[i][j]为0,显然有f[i][j]=0
如果不为0,那么f[i][j]=min(f[i-1][j-1],f[i][j-1],f[i-1][j])+1
为什么是这样的?
我们可以由这个点的上点,左点,左上点来转移它的状态
由于正方形中不能有0,所以转移时要用最小的状态来转移(最大的话一定会有0混入正方形)
#include <cstdio>
#include <iostream>
using namespace std;
int map[1000][1000];
int f[1000][1000];
int screen(int n,int m)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
if(map[i][j])
f[i][j]=min(f[i-1][j-1],min(f[i-1][j],f[i][j-1]))+1;
int maxf=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
maxf=max(maxf,f[i][j]);
return maxf;
}
int main()
{
int n,m;
int ans=1;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&map[i][j]);
printf("%d",screen(n,m));
return 0;
}