c++单调栈实现最大矩形面积

题意

给一个直方图,求直方图中的最大矩形的面积。例如,下面这个图片中直方图的高度从左到右分别是2, 1, 4, 5, 1, 3, 3, 他们的宽都是1,其中最大的矩形是阴影部分。
在这里插入图片描述

Input

输入包含多组数据。每组数据用一个整数n来表示直方图中小矩形的个数,你可以假定1 <= n <= 100000. 然后接下来n个整数h1, …, hn, 满足 0 <= hi <= 1000000000. 这些数字表示直方图中从左到右每个小矩形的高度,每个小矩形的宽度为1。 测试数据以0结尾。

Output

对于每组测试数据输出一行一个整数表示答案。

Sample Input

7 2 1 4 5 1 3 3
4 1000 1000 1000 1000
0

Sample Output

8
4000

单调栈理解

单调栈=单调+栈;
单调性: 只可意会不可言传 擅用搜索引擎 ;
栈:先进后出的一种线性数据结构(更多内容参考搜索引擎);
单调栈,有序的栈(一定是从栈顶到栈底满足单调性),push过程中删除破坏有序结构的数据(菜鸡理解 );
首先来看一下单调栈的性质:
单调栈的类型:

单调递增栈(从栈顶到栈底满足递增)
单调递减栈(从栈顶到栈底满足递减)

单调栈的作用:

线性的时间复杂度:每个数据只入栈一次,出栈一次,所以O(n)的时间复杂度
单调递
栈:可以找到从当前元素往左/往右第一个比当前元素的元素;
单调递栈:可以找到从当前元素往左/往右第一个比当前元素的元素;
可以求得以当前元素为最值得最大区间(结合上面两条自行理解)。

用单调递减栈举个栗子(递增栈类似):
有一组数据:

2 1 4 5 1 3 3
以第一个元素2为栗,右边第一个比2小的元素为 1(下标为2),左边没有,所以为0;
以中间元素5为栗,左边第一个比5小的元素为4,右边第一个比5小的元素为1(下标为5)
类似的,可以求出其他元素的往左/往右的第一个小于当前元素的值。

在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
到这我们再看数列:
在这里插入图片描述
在这里插入图片描述
我们发现,被弹出元素右边第一个比自己小的元素就是弹出自己的元素。
反过来入栈,我们即可求出左边比自己小的元素。

思路

最大的矩形面积,我们只需要找到当前高度可以延展的最大宽度即可。
用题目中的数据举例:

1, 4, 5, 1, 3, 3

1可以向左向右延展的最大区间为[1,6],所以以1为高度,面积为6×1=6;
4可以向左向右延展的最大区间为[2,3],因为1和4都比它低 ,不能延展出去,换句话说,只有两边比它高的才能延展出去,所以以4为高度,面积为4×2=8;
…(剩下的自己求一下)
最后找出最大面积即可。

代码:

#include<iostream>
#include <cstdio>
using namespace std;
int n;
long long l[100010]= {0};
long long r[100010]= {0};
long long a[100010];
long long st[100010];  //区分相同元素 所以存储下标  数组模拟栈
void getRight()  //得到每一个右边元素比自己小的元素下标
{//模拟单调栈的规则
	int t=0;
	for(int i=n-1; i>=0; i--)
	{
		while(t>0&&a[i]<=a[st[t-1]])
		{
			t--;
		}
		if(t==0)   //说明有边界为栈底 可以延展到最后
			r[i]=n; 
		else
			r[i]=st[t-1];
		st[t++]=i;
	}
}
void getLeft()   //得到每一个左边元素比自己小的元素下标
{
	int t=0;
	for(int i=0; i<n; i++)
	{
		while(t>0&&a[i]<=a[st[t-1]])
		{
			t--;
		}
		if(t==0)
			l[i]=0;   //说明有边界为栈顶 可以延展到最开始
		else
			l[i]=st[t-1]+1; 
		st[t++]=i;
	}
}
int main()
{
	while(1)
	{
		scanf("%d",&n);
		if(n==0) break;
		for(int i=0; i<n; i++)
		{
			scanf("%lld",&a[i]);
		}
		getLeft();
		getRight();
		long long ans = 0;
		for (int i = 0; i <n; i++)
		{
			ans=max(ans,(r[i] - l[i] ) * a[i]);		
		}
		cout<<ans<<endl;
	}

	return 0;
}

我们进一步探讨:

2 1 4 5 1 3 3

这组数据有7个数,刚才我们只找到了3个往右的值。
现在我们假设,在数据最开始和末尾各放一个零,那么我们是不是就可以找到所有元素的往左/往右的第一个比它小的值,即找到0,就意味着可以往左/往右没有比它小的元素。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
到这
在这里插入图片描述
我们发现,一个元素,被谁弹出来,谁就是其右边的第一个比它小的元素,
这个元素在栈内右边的元素,是其左边的第一个比它小的元素。
(如果有两个相同元素,则只有一个范围正确,第二个范围 有点偏差
(上图使用条件是栈内不严格单调,使用严格单调则正确的那个值会返回反过来,总之只有一个正确,也可能是我的推导错了,欢迎指正 )
用这个方法,我们只需遍历一边即可拿到左右边界值。

代码

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include<stack>
#include<algorithm>
#define ll long long 
using namespace std;
ll a[100005];
ll r[100005];
ll l[100005];

int main()
{
	int n;
	while (1)
	{
		scanf("%d", &n);
		if (n == 0) return 0;
		for (int i = 1; i <= n; i++)
		{
			scanf("%lld", &a[i]);
		}
		a[n + 1] = 0; //最后一个0
		stack<int> st;
		st.push(0);  //第一个0
		for (int i = 1; i <= n + 1; i++)
		{
			while (!st.empty() && a[i] < a[st.top()]) 
			{
				r[st.top()] = i;    
				//被谁弹出来,谁就是其右边的第一个比它小的元素
				st.pop();
			}
			if(!st.empty()) l[i] = st.top();   //入栈前的栈顶是他的右边元素
			//栈内右边的元素,是其左边的第一个比它小的元素
			st.push(i);
		}
		long long ans = 0;
		for (int i = 1; i <= n; i++)
		{
			ans = max(ans, (r[i] - l[i]-1) * a[i]);
		}
		cout << ans << endl;
	}
	return 0;
}

总结:

这道题得开long long 要不然过不去。
一定要先用手过一遍,看看栈内是严格单调还是不严格单调。

  • 6
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值