BJTUOJ 1851 wd 打 boss（dp+单调栈）

Description

众所周知，$wd$非常喜欢玩游戏，最近他在玩一款游戏的时候卡在了最后一关。

最后一关有 $n×n$ 块地板，$n$行$n$列，每块地板上都有尖刺，初始的时候一些地板的尖刺朝上，一些地板的尖刺朝下，尖刺朝上的地板不可以踩。$wd$可以通过使用氪金道具翻转一整行或者一整列的地板，使原来朝上的尖刺变成朝下，朝下的尖刺变成朝上。

现在 $wd$ 需要找到一块最大的没有尖刺的矩形区域当作他和最终 $boss$ 的决战场地，他有无限多的氪金道具可以用，请你帮他算一算他能找到面积最大的决战场地有多大。

Input

第一行为一个整数$n (1\le n\le 500)$

接下来有 $n$ 行数据，每行有$n$ 个整数，第 $i$ 行的第 $j$ 个数代表第 $i$ 行第 $j$列的地板的初始状态，$0$代表尖刺朝下，$1$代表尖刺朝上。

Output

决战场地的最大面积。

Sample Input

2
0 1
1 0

Sample Output

4

Solution

首先对行列的操作顺序不重要，先考虑对行的操作，一行是否操作并不影响这一行相邻元素的关系（相等或不等），而对一列做操作会同时改变这一列与相邻列元素的关系，故若想这一个子矩阵全部是$0$，那么相邻列的关系必然是全部相等或全部不等

枚举矩阵下边$i$，维护$h[i][j][0/1]$表示从第$i$行第$j$列的元素开始沿着第$j$列往上和第$i$行第$j+1$个元素开始沿第$j+1$列向上连续相等/不等的元素长度，以$a[j]=max(h[i][j][0],h[i][j][1]),1\le j<n$，以第$i$行为底边所能取到的合法子矩阵的高度必然是某个$a[j]$值，故用单调栈维护$a[j]$在序列中作为最小值的最大存在区间$[l[j],r[j]]$，那么第$i$行为底边的合法子矩阵最大规模即为$max\{a[j]\cdot (r[j]-l[j]+2),1\le j<n\}$，因为区间长度为$r[j]-l[j]+1$表示有相邻的$r[j]-l[j]+2$列满足公共部分高度为$a[j]$，维护该最大值即为答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=505;
int n,s[maxn][maxn],h[maxn][maxn][2]; 
int a[maxn],l[maxn],r[maxn],sta[maxn],p;
int Solve(int n)
{
    for(int i=1;i<=n;i++)l[i]=r[i]=i;
    p=0;
    for(int i=1;i<=n;i++)
    {
        if(!p||a[i]>=a[sta[p]])sta[++p]=i;
        else 
        {
            while(p&&a[i]<a[sta[p]])
                r[sta[p]]=i-1,p--;
            sta[++p]=i;
        }
    }
    while(p)r[sta[p]]=n,p--;
    for(int i=n;i>=1;i--)
    {
        if(!p||a[i]>=a[sta[p]])sta[++p]=i;
        else 
        {
            while(p&&a[i]<a[sta[p]])
                l[sta[p]]=i+1,p--;
            sta[++p]=i;
        }
    }
    while(p)l[sta[p]]=1,p--;
    int ans=0;
    for(int i=1;i<=n;i++)ans=max(ans,(r[i]-l[i]+2)*a[i]);
    return ans;
} 
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&s[i][j]);
        if(n==1)
        {
            printf("1\n");
            continue;
        }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=2;j<=n;j++)
                if(s[i][j]==s[i][j-1])h[i][j][1]=h[i-1][j][1]+1,h[i][j][0]=0;
                else h[i][j][0]=h[i-1][j][0]+1,h[i][j][1]=0;
            for(int j=2;j<=n;j++)a[j-1]=max(h[i][j][0],h[i][j][1]);
            ans=max(ans,Solve(n-1));
        }
        printf("%d\n",ans);
    }
    return 0;
}