[BZOJ]1057: [ZJOI2007]棋盘制作 悬线法求最大子矩阵

Description

　　国际象棋是世界上最古老的博弈游戏之一，和中国的围棋、象棋以及日本的将棋同享盛名。据说国际象棋起源

于易经的思想，棋盘是一个8*8大小的黑白相间的方阵，对应八八六十四卦，黑白对应阴阳。而我们的主人公小Q，

正是国际象棋的狂热爱好者。作为一个顶尖高手，他已不满足于普通的棋盘与规则，于是他跟他的好朋友小W决定

将棋盘扩大以适应他们的新规则。小Q找到了一张由N*M个正方形的格子组成的矩形纸片，每个格子被涂有黑白两种

颜色之一。小Q想在这种纸中裁减一部分作为新棋盘，当然，他希望这个棋盘尽可能的大。不过小Q还没有决定是找

一个正方形的棋盘还是一个矩形的棋盘（当然，不管哪种，棋盘必须都黑白相间，即相邻的格子不同色），所以他

希望可以找到最大的正方形棋盘面积和最大的矩形棋盘面积，从而决定哪个更好一些。于是小Q找到了即将参加全

国信息学竞赛的你，你能帮助他么？

Input

　　第一行包含两个整数N和M，分别表示矩形纸片的长和宽。接下来的N行包含一个N * M的01矩阵，表示这张矩形

纸片的颜色（0表示白色，1表示黑色）。

Output

　　包含两行，每行包含一个整数。第一行为可以找到的最大正方形棋盘的面积，第二行为可以找到的最大矩形棋

盘的面积（注意正方形和矩形是可以相交或者包含的）。

题解：经典的最大子矩阵问题，我一开始想的是对于每个点，分别记录向上向下最多能延伸到哪里，然后从左往右扫描，更新矩形的长宽，无限TLE……后来学了悬线法求最大子矩阵（详细的介绍可以去看王知昆的国家集训队论文）。大概思路：up[i][j]表示点(i,j)向上最多能延伸到哪里（假设这个最远能延伸到的点为x），L[i][j]表示x到(i,j)的这条竖线向左最多能延伸到哪里，R[i][j]表示x到(i,j)的这条竖线向右最多能延伸到哪里。对于每个点，都可以更新矩形的最大面积ans=max(ans,(R[i][j]-L[i][j]+1)*up[i][j])，对于正方形，它一定在一个矩形里面，而每一个点的up、L、R都能根据上方的点O（1）求出，所以时间复杂度为O（nm）。听说还可以用单调栈做？蒟蒻不会啊！

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=2005;
int n,m,a[maxn][maxn],up[maxn][maxn],L[maxn][maxn],R[maxn][maxn],ans1=1,ans2=1;
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]),L[i][j]=R[i][j]=j,up[i][j]=1;
    for(int i=1;i<=n;i++)
    for(int j=2;j<=m;j++)
    if(a[i][j]==1-a[i][j-1]) L[i][j]=L[i][j-1];
    for(int i=1;i<=n;i++)
    for(int j=m-1;j>0;j--)
    if(a[i][j]==1-a[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 && a[i][j]==1-a[i-1][j])
        {
        	L[i][j]=max(L[i][j],L[i-1][j]);
        	R[i][j]=min(R[i][j],R[i-1][j]);
        	up[i][j]=up[i-1][j]+1;
		}
		int A=R[i][j]-L[i][j]+1,B=min(A,up[i][j]);
		ans1=max(ans1,B*B);
		ans2=max(ans2,A*up[i][j]);
    }
    printf("%d\n%d",ans1,ans2);
}