单调栈-C++实现

本文介绍了单调栈这种算法数据结构,用于解决最大矩形面积和每日温度变化的问题。通过维护单调递增或递减的栈,有效地计算出问题的解,实现了空间换时间的优化。文中给出了两个具体实例:一是求直方图中最大矩形面积,二是求每日气温中等待更高温度的天数。
摘要由CSDN通过智能技术生成

单调栈

本人小学鸡,写博客仅为强化记忆,分享交流,如有错误请多指正。

何为单调栈?
顾名思义,就是单调的栈,我们维护一个栈,栈中的元素保持非严格单调递增或递减。
此算法在处理一些问题时莫名其妙(雾)的有效,是一个典型的空间换时间的算法。

1. 最大的矩形 [CSP]201312-3 / [LeetCode] 84.

问题描述
在横轴上放了n个相邻的矩形,每个矩形的宽度是1,而第i(1 ≤ i ≤ n)个矩形的高度是hi。这n个矩形构成了一个直方图。例如,下图中六个矩形的高度就分别是3, 1, 6, 5, 2, 3。在这里插入图片描述

请找出能放在给定直方图里面积最大的矩形,它的边要与坐标轴平行。对于上面给出的例子,最大矩形如下图所示的阴影部分,面积是10。
  在这里插入图片描述

输入格式
第一行包含一个整数n,即矩形的数量(1 ≤ n ≤ 1000)。
第二行包含n 个整数h1, h2, … , hn,相邻的数之间由空格分隔。(1 ≤ hi ≤ 10000)。hi是第i个矩形的高度。
  
输出格式
输出一行,包含一个整数,即给定直方图内的最大矩形的面积。
  
样例输入
6
3 1 6 5 2 3

样例输出
10

题解
我们可以维持一个单调递增的栈,为了便于计算矩形宽度,我们在栈里存放单个矩形的位置。
我们从左到右遍历高度数组,对于每个矩形的高度p,如果p大于等于当前栈顶储存位置的高度q,我们将p的位置也压入栈中;
如果p小于q,我们将q弹出,纪录高度,并记录当前遍历到的矩形p与新栈顶位置之差(实际上还需要减1),作为宽度,并更新结果。
为避免栈中剩余矩形,我们可以在数组尾插入一个高度为零的矩形,使栈中所有矩形弹出并更新。

代码实现

int largestRectangleArea(vector<int>& heights) {
	heights.push_back(0);		//插入空矩形,弹出栈中剩余矩形
	int len = heights.size(), area = 0, pre_index, height, width;
	stack<int> indices;
	for (int i = 0; i < len; i++) {
		while (!indices.empty() && heights[indices.top()] > heights[i]) {	//检查栈是否为空
			pre_index = indices.top();		//储存栈顶矩形的位置
			indices.pop();
			height = heights[pre_index];	//储存高度
			if (indices.empty()) {			//避免操作空栈
				width = i;					//若弹出至栈为空,因栈的递增性,边界可向左延伸至0
			} else {
				width = i - indices.top() - 1;		//储存宽度
			}
			area = area > (width * height) ? area : (width * height);		//更新结果
		}
		indices.push(i);
	}
	return area;
}

2.每日温度 [LeetCode] 739.

问题描述

请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

题解

我们在这里维持一个单调递减的栈,为了便于计算天数差,我们在栈中储存位置(即日期)。
此题相较上题简单一些,故不在此赘述。
注意:
此题与上题不同的是,若栈中有剩余的日期,说明此后没有更暖和的日期,故无需再弹出。

代码实现

vector<int> dailyTemperatures(vector<int>& T) {
	int n = T.size(), pre_index;
	vector<int> ans(n);
	stack<int> indices;
	for (int i = 0; i < n; i++) {
		while (!indices.empty() && T[indices.top()] < T[i]) {
			pre_index = indices.top();
			indices.pop();
			ans[pre_index] = i - pre_index;		//计算日期差
		}
		indices.push(i);
	}
	return ans;
}
  • 7
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值