[ZJOI2007] 棋盘制作（悬线法好题）

[ZJOI2007] 棋盘制作

题目描述

国际象棋是世界上最古老的博弈游戏之一，和中国的围棋、象棋以及日本的将棋同享盛名。据说国际象棋起源于易经的思想，棋盘是一个 $8\times 8$ 大小的黑白相间的方阵，对应八八六十四卦，黑白对应阴阳。

而我们的主人公 小Q，正是国际象棋的狂热爱好者。作为一个顶尖高手，他已不满足于普通的棋盘与规则，于是他跟他的好朋友 小W 决定将棋盘扩大以适应他们的新规则。

小Q 找到了一张由 $N\times M$ 个正方形的格子组成的矩形纸片，每个格子被涂有黑白两种颜色之一。小Q 想在这种纸中裁减一部分作为新棋盘，当然，他希望这个棋盘尽可能的大。

不过 小Q 还没有决定是找一个正方形的棋盘还是一个矩形的棋盘（当然，不管哪种，棋盘必须都黑白相间，即相邻的格子不同色），所以他希望可以找到最大的正方形棋盘面积和最大的矩形棋盘面积，从而决定哪个更好一些。

于是 小Q 找到了即将参加全国信息学竞赛的你，你能帮助他么？

输入格式

包含两个整数 $N$ 和 $M$，分别表示矩形纸片的长和宽。接下来的 $N$ 行包含一个 $N\times M$ 的 $01$ 矩阵，表示这张矩形纸片的颜色（$0$ 表示白色，$1$ 表示黑色）。

输出格式

包含两行，每行包含一个整数。第一行为可以找到的最大正方形棋盘的面积，第二行为可以找到的最大矩形棋盘的面积（注意正方形和矩形是可以相交或者包含的）。

样例 #1

样例输入 #1

3 3
1 0 1
0 1 0
1 0 0

样例输出 #1

4
6

提示

对于 $20\%$ 的数据，$N,M\le 80$

对于 $40\%$ 的数据，$N,M\le 400$

对于 $100\%$ 的数据，$N,M\le 2000$

代码

//只需要掌握悬线法的原理，这是一道非常好的最大子矩阵面积的题目

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>

#define int long long

using namespace std;

const int N = 2010;

int w[N][N];
int l[N][N],r[N][N],h[N][N];
int n,m;
int res1,res2;

int read(){
    int x=0;
    
    char c=getchar();
    
    while(!(c>='0'&&c<='9')){
        c=getchar();
    }
    
    while(c>='0'&&c<='9'){
        x=(x<<3)+(x<<1)+c-'0';
        c=getchar();
    }
    
    return x;
}

signed main(){
    
    n=read(),m=read();
    memset(w,-1,sizeof w);
    
    // cout<<n<<' '<<m<<endl;
    
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            // cin>>w[i][j];
            w[i][j]=read();
            l[i][j]=r[i][j]=j;
        }
    }
    
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(w[i][j]^w[i][j+1]){
                h[i][j]=1;
            }
        }
    }
    
    //每行
    for(int i=1;i<=n;i++){
        for(int j=2;j<=m;j++){//处理左右
        //处理左
            if(w[i][j]^w[i][j-1])l[i][j]=l[i][j-1];
        }
        //处理右
        for(int j=m-1;j>=1;j--){
            if(w[i][j]^w[i][j+1])r[i][j]=r[i][j+1];
        }
    }
    
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            if(i>1&&w[i][j]^w[i-1][j]){
                h[i][j]=h[i-1][j]+1;
                l[i][j]=max(l[i][j],l[i-1][j]);
                r[i][j]=min(r[i][j],r[i-1][j]);
                res1=max(res1,h[i][j]*(r[i][j]-l[i][j]+1));
                int k=min(r[i][j]-l[i][j]+1,h[i][j]);
                res2=max(res2,k*k);
            }
        }
    }
    cout<<res2<<endl;
    cout<<res1<<endl;
    
    return 0;
    
}