单调栈求01矩阵的数量和最大面积

_hh123_

已于 2022-07-29 11:12:06 修改

阅读量171

点赞数

分类专栏：单调栈文章标签：算法 c++

于 2022-07-27 09:39:05 首次发布

本文链接：https://blog.csdn.net/weixin_51799916/article/details/126007453

版权

单调栈专栏收录该内容

1 篇文章 0 订阅

订阅专栏

题目链接：P1950长方形

题目描述

简化就是给定一个只有0和1的 $n * m$ 矩阵，求有多少子矩阵且子矩阵中只包含0或者1。

【数据规模】
$1\leq n\leq 1000,1\leq m\leq 1000$

这道题的数据范围只允许我们用 $O (n * m)$ 的时间复杂度。这里以求只包含 $1$ 的子矩阵为例
为了防止在找矩阵的时候出现找重的情况，我们可以找矩阵的底边来找矩阵的数量。
首先定义三个数组分别是：

$h [i] [j]$ ：为第 $i$ 行第 $j$ 列可向上延伸多少(当前位置的数 $1$ ，往上有多少个 $1$ )
注意的是当前位置的数如果是 $0$ ，那么h[i][j]也要是 $0$

$l [i] [j]$ ：第 $i$ 行第 $j$ 列从 $j$ 往左第一个小于等于 $h [i] [j]$ 的数的下标

$r [i] [j]$ ：第 $i$ 行第 $j$ 列从 $j$ 往右第一个小于 $h [i] [j]$ 的数的下标

这里维护 $l$ 和 $r$ 的时候可以利用单调栈来优化

以第i行为底边的矩阵数量为： $_{j=1}^{m}(j-l[i][j])*(r[i][j]-j)*h[i][j];$

公式推导：我们定义的 $h$ 就是利用这个高度来限制形成以第 $i$ 行 $j$ 列高度的矩形能往左往右延伸到哪里，往左可以取 $j - l [i] [j]$ 个，往右可以取到 $r [i] [j] - j$ 个，高度有 $h [i] [j]$ 种，利用乘法原理即可得出。

画个图就很容易理解公式以及为什么一个是小于等于，另一个是严格小于，是为了防止两个高度相等且相邻的情况，造成重复计算。
具体证明：长方形题解
代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#define x first
#define y second
using namespace std;
const int N=1010;
char g[N][N];
int h[N][N],l[N][N],r[N][N];
//L:第i行第j列从j往左第一个小于等于h[i][j]的数的下标
//R:第i行第j列从j往右第一个小于h[i][j]的数的下标
int n,m;
pair<int,int> stk[N];
void solve()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>g[i]+1;
    for(int j=1;j<=m;j++)
    {
        int pos=1;
        for(int i=1;i<=n;i++)
        {
            if(g[i][j]=='*')
            {
                h[i][j]=0;
                pos=i+1;
            }
            else
                h[i][j]=i-pos+1;
        }
    }
    /*for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        cout<<h[i][j]<<" ";
        cout<<endl;
    }
    cout<<endl;*/
    for(int i=1;i<=n;i++)
    {
        int top=0;
        stk[++top]={-1,0};//num index
        for(int j=1;j<=m;j++)
        {
            while(top&&stk[top].x>h[i][j]) top--;
            l[i][j]=stk[top].y;
            stk[++top]={h[i][j],j};
        }
    }
   /* for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        cout<<l[i][j]<<" ";
        cout<<endl;
    }*/
    for(int i=1;i<=n;i++)
    {
        int top=0;
        stk[++top]={-1,m+1};//num index
        for(int j=m;j>=1;j--)
        {
            while(top&&stk[top].x>=h[i][j]) top--;
            r[i][j]=stk[top].y;
            stk[++top]={h[i][j],j};
        }
    }
    /*for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        cout<<r[i][j]<<" ";
        cout<<endl;
    }*/
    long long res=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            res+=(j-l[i][j])*(r[i][j]-j)*h[i][j];
    }
    cout<<res<<endl;
}
int main()
{
    int t;
//	cin>>t;
	t=1;
	while(t--)
	{
		solve();
	}
    return 0;
}

上面的分析可以求出来矩阵的数量，在此基础上如何求只包含1的子矩阵的最大面积。
链接：HDU1505
题意类似，结果为最大面积 $* 3$
问题关键是求出最大面积，上面的公式推导我们可以不重不漏的得到所有矩阵的方案数，那么在此基础上，改成求最大面积即可。即： $res = ma x (res, (r [i] [j] - l [i] [j] - 1) * h [i] [j])$
为什么不直接用 $(j - l [i] [j]) + (r [i] [j] - j)$ 呢，因为左侧和右侧都包含了第 $j$ 列，需要 $- 1$ 去掉重复计算的长度。
代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#define x first
#define y second
using namespace std;
const int N=1010,M=N;
char g[N][M];
int h[N][M],l[N][M],r[N][M];
int n,m;
pair<int,int> stk[N];
void solve()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        cin>>g[i][j];
    for(int j=1;j<=m;j++)
    {
        int pos=1;
        for(int i=1;i<=n;i++)
        {
            if(g[i][j]=='R')
            {
                h[i][j]=0;
                pos=i+1;
            }
            else
                h[i][j]=i-pos+1;
        }
    }
    /*for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        cout<<h[i][j]<<" ";
        cout<<endl;
    }
    cout<<endl;*/
    for(int i=1;i<=n;i++)
    {
        int top=0;
        stk[++top]={-1,0};//num index
        for(int j=1;j<=m;j++)
        {
            while(top&&stk[top].x>h[i][j]) top--;
            l[i][j]=stk[top].y;
            stk[++top]={h[i][j],j};
        }
    }
   /* for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        cout<<l[i][j]<<" ";
        cout<<endl;
    }*/
    for(int i=1;i<=n;i++)
    {
        int top=0;
        stk[++top]={-1,m+1};//num index
        for(int j=m;j>=1;j--)
        {
            while(top&&stk[top].x>=h[i][j]) top--;
            r[i][j]=stk[top].y;
            stk[++top]={h[i][j],j};
        }
    }
    /*for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        cout<<r[i][j]<<" ";
        cout<<endl;
    }*/
    int res=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
            res=max(res,(r[i][j]-l[i][j]-1)*h[i][j]);
    }
    cout<<(long long)res*3<<endl;
}
int main()
{
    int t;
    t=1;
	cin>>t;
	for(int i=1;i<=t;i++)
	{
		solve();
	}
    return 0;
}