洛谷 P1169 [ZJOI2007]棋盘制作

题目链接
P1169 [ZJOI2007]棋盘制作
题目描述
国际象棋是世界上最古老的博弈游戏之一，和中国的围棋、象棋以及日本的将棋同享盛名。据说国际象棋起源于易经的思想，棋盘是一个8×8大小的黑白相间的方阵，对应八八六十四卦，黑白对应阴阳。
而我们的主人公小Q，正是国际象棋的狂热爱好者。作为一个顶尖高手，他已不满足于普通的棋盘与规则，于是他跟他的好朋友小W决定将棋盘扩大以适应他们的新规则。
小Q找到了一张由N×M个正方形的格子组成的矩形纸片，每个格子被涂有黑白两种颜色之一。小Q想在这种纸中裁减一部分作为新棋盘，当然，他希望这个棋盘尽可能的大。
不过小Q还没有决定是找一个正方形的棋盘还是一个矩形的棋盘（当然，不管哪种，棋盘必须都黑白相间，即相邻的格子不同色），所以他希望可以找到最大的正方形棋盘面积和最大的矩形棋盘面积，从而决定哪个更好一些。
于是小Q找到了即将参加全国信息学竞赛的你，你能帮助他么？

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

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

输入样例

3 3
1 0 1
0 1 0
1 0 0

输入样例

4
6

left[ i ][ j ] 表示从方格（i，j）往左所能到达的最左边的格子的纵坐标
right[ i ][ j ] 表示从方格（i，j）往右所能到达的最右边的格子的纵坐标
up[ i ][ j ] 表示从方格（i，j）往上所能到达的最高的格子的横坐标
先算出每一个格子在当前行向左向右可以到达最远的纵坐标（高度为1的矩阵）
然后从上到下从左到右遍历每一个方格（i，j），计算包含每一个方格高度优先的满足条件的矩阵的面积，更新最大值。

代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <cctype>
#include <sstream>
#define LL long long
#define _for(i,j,k) for(int i=j;i<=k;i++)
#define for_(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
const int maxn = 1e3+5;
int ta[maxn][maxn],l[maxn][maxn],r[maxn][maxn],up[maxn][maxn],n,m; 
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    cin>>n>>m;
    _for(i,1,n)
    _for(j,1,m){
        cin>>ta[i][j];
        l[i][j]=r[i][j]=j;
        up[i][j]=1;
    }
    _for(i,1,n)
    _for(j,2,m) if(ta[i][j]!=ta[i][j-1]){//初始化l[i][j]
        l[i][j]=l[i][j-1];
    }
    _for(i,1,n)
    for_(j,m-1,1) if(ta[i][j]!=ta[i][j+1]){//初始化r[i][j]
        r[i][j]=r[i][j+1];
    }
    int a=0,b,an1=0,an2=0;
    _for(i,1,n)
    _for(j,1,n){
        if(i>1&&ta[i][j]!=ta[i-1][j]){//格子(i,j)可以向上扩展
            l[i][j]=max(l[i][j],l[i-1][j]);//包含格子(i,j)的矩形的左边界
            r[i][j]=min(r[i][j],r[i-1][j]);//包含格子(i,j)的矩形的右边界
            up[i][j]=up[i-1][j]+1;//上边界（下边界为i）
        }
        a=r[i][j]-l[i][j]+1;//矩形的长，up[i][j]为矩形的宽/高
        b=min(a,up[i][j]);//短边
        an1=max(an1,b*b);//正方形面积
        an2=max(an2,a*up[i][j]);//矩形面积
    }
    cout<<an1<<"\n"<<an2;
    return 0;
}