编程语言
首页 > 编程语言> > LeetCode力扣84.柱状图中最大的矩形(C++)【单调栈】详细解析+代码注释

LeetCode力扣84.柱状图中最大的矩形(C++)【单调栈】详细解析+代码注释

作者:互联网

LeetCode力扣84.柱状图中最大的矩形

题目描述

给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。求在该柱状图中,能够勾勒出来的矩形的最大面积。

image1

输入输出样例

输入 #1

6
2 1 5 6 2 3

输出 #1

10

解释:最大的矩形为图中红色区域,面积为 10

解题思路

首先第一眼可以看出本题用暴力法一定可以做出来,但纯暴力一般会超时,可以借助单调栈对暴力法进行优化。

先说暴力法,最外层循环遍历 n 根柱子,内部对每根柱子左右分别进行 while 循环找到第一根高度小于当前高度的柱子,就可以算出以当前柱子为高时矩形的最大面积,遍历结束即可找出矩形最大面积。

单调栈法,依旧是最外层循环遍历 n 根柱子,并且只在栈中存储高度递增的柱子的数组下标。比如第 i 根柱子的高度大于栈顶的柱子高度时,就让i入栈,小于栈顶柱子高度,说明以栈顶为高的矩形找到了右边界,而栈中存储的是递增的序列,那么左边界也确定了,左右边界做差就得到了矩形的宽,那么以栈顶为高的矩形的最大面积就确定了,将面积与之前记录的最大面积进行比较,更新最大面积。遍历结束就得到了最终结果。这样就省去了纯暴力中每次循环查找左右边界的过程。光看文字看不懂,这里建议看代码画图自己模拟一遍。

要注意的是最外层遍历要多执行一次,for(int i=0;i<=n;i++),下标为 n 的数组中存柱子高度为0,因为如果出现 n 根柱子高度恰好为递增序列的情况,那么程序不会对最大面积进行更新。

代码

此代码不支持上机检测。

#include<iostream>
#include<algorithm>
#include<stack>

using namespace std;

int n;	//柱子数量 
int max_area;	//记录当前最大面积 
int area;	//记录以当前柱子为高的最大面积 
int height[100005];	//存储柱子高度 
stack<int> s;	//存储高度递增的柱子的数组下标 

int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
		cin>>height[i];
	for(int i=0;i<=n;i++)
	{
		while(!s.empty()&&height[s.top()]>height[i])	//栈非空且第i根柱子高度栈顶柱子高度,此时不能入栈,开始计算面积 
		{
			if(s.size()==1) area=height[s.top()];	//当栈中仅有一个元素,矩形高为柱子高度宽为1 
			else area=height[s.top()]*(i-s.top());	//当栈中有多个元素,矩形高为栈顶柱子高度,宽为左右第一个比当前柱子矮的两柱子的间距即i-s.top 
			max_area=max(max_area,area);	//与之前记录的最大面积作比较,更新最大面积 
			s.pop();	//已做过高的柱子出栈 
		}
		s.push(i);	//栈空或者栈顶柱子高度小于第i根柱子高度,下标入栈 
	}
	cout<<max_area;
	return 0;
}

标签:柱子,area,int,面积,高度,C++,力扣,柱状图,矩形
来源: https://www.cnblogs.com/LoginX/p/Login_X3.html