bzoj 1057: [ZJOI2007]棋盘制作 （dp+悬线法）

1057: [ZJOI2007]棋盘制作

Time Limit: 20 Sec   Memory Limit: 162 MB
Submit: 2465   Solved: 1220
[ Submit][ Status][ Discuss]

Description

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

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

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

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

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

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

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

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

Input

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

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

Output

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

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

Sample Input

3 3
1 0 1
0 1 0
1 0 0

Sample Output

4
6

HINT

N, M ≤ 2000

Source

[ Submit][ Status][ Discuss] 

题解：dp+悬线法。

h[i][j] 表示从(i,j)可以向上延伸的最大距离。

pre[i][j]表示从(i,j)可以向左延伸的最大距离。

next[i][j]表示从(i,j)可以向右延伸的最大距离。

每次只有满足限制条件才更新：h[i][j]=h[i-1][j]+1 pre[i][j]=max(pre[i][j],pre[i-1][j])

next[i][j]=min(next[i][j],next[i-1][j])

求最大矩形的时候用h[i][j]*(next[i][j]-pre[i][j]+1)更新答案。

因为最大正方形一定包含在某个满足条件的极大子矩形中，所以每次更新答案的时候顺便更新一下正方形的情况即可。

#include<iostream>
#include<cstdio> 
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 2003
using namespace std;
int n,m,ans,ans1;
int a[N][N],pre[N][N],next[N][N];
int l[N][N],r[N][N],h[N][N];
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]),pre[i][j]=next[i][j]=j,h[i][j]=1;
    for (int i=1;i<=n;i++)
     for (int j=2;j<=m;j++)
      if (a[i][j]==a[i][j-1]^1) pre[i][j]=pre[i][j-1];
    for (int i=1;i<=n;i++)
     for (int j=m-1;j>=1;j--) 
      if (a[i][j]==a[i][j+1]^1) next[i][j]=next[i][j+1];
    for (int i=1;i<=m;i++) pre[0][i]=0,next[0][i]=m+1;
    for (int i=1;i<=n;i++)
     for (int j=1;j<=m;j++)
     {
      if (a[i][j]==a[i-1][j]^1&&i!=1) {
        h[i][j]+=h[i-1][j];
        pre[i][j]=max(pre[i][j],pre[i-1][j]);
        next[i][j]=min(next[i][j],next[i-1][j]);
      }
      int t=min(h[i][j],next[i][j]-pre[i][j]+1);
      //cout<<h[i][j]<<" "<<next[i][j]<<" "<<pre[i][j]<<" "<<j<<endl;
      ans=max(ans,t*t);
      ans1=max(ans1,h[i][j]*(next[i][j]-pre[i][j]+1));
     }
    printf("%d\n%d\n",ans,ans1);
}