P1681 最大正方形II-动态规划

最大正方形II

题目背景

忙完了学校的事，v神终于可以做他的“正事”：陪女朋友散步。一天，他和女朋友走着走着，不知不觉就来到了一个千里无烟的地方。v神正要往回走，如发现了一块牌子，牌子上有有一行小字和一张图，小字说道：“找到图上最大的交错正方形之后和我联系，这块地就是你的了。”在房价疯长的年代，v神当然不愿错过这个机会，于是开始找了起来……以v神的能力当然找不出来了，你能帮v神找出来吗？

题目描述

图上有一个矩阵，由N*M个格子组成，这些格子由两种颜色构成，黑色和白色。请找到面积最大的且内部是黑白交错（即两个相连的正方形颜色不能相同）的正方形。

输入格式

第一行两个整数N和M，分别表示行数和列数。接下来有N行，每行M个数，0或1分别表示这个格子是黑色或白色。

输出格式

仅有一行，表示满足条件最大正方形的 边长

样例 #1

样例输入 #1

3 3
0 1 0
1 0 0
1 1 1

样例输出 #1

2

#include <bits/stdc++.h>
using namespace std;
#define de(x) cout<<x<<" ";
#define Pu puts("");
#define sf(x) scanf("%d",&x);
const int N=2e3+10;
int n,m,ans;
int a[N][N];
int f[N][N][2];
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            sf(a[i][j])
            if(i==1||j==1) f[i][j][0]=f[i][j][1]=1;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(a[i][j]==1) f[i][j][1]=min(f[i-1][j-1][1],min(f[i-1][j][0],f[i][j-1][0]))+1,ans=max(ans,f[i][j][1]);
            if(a[i][j]==0) f[i][j][0]=min(f[i-1][j-1][0],min(f[i-1][j][1],f[i][j-1][1]))+1,ans=max(ans,f[i][j][0]);
        }
    }
    de(ans)
    return 0;
}