单调栈其实就是栈,只是运用了一些巧妙的逻辑,使得每次入栈的元素都能保持有序(单调递增或者单调递减)
单调栈模板
现在给你出这么⼀道题: 给你⼀个数组 nums,请你返回⼀个等长的结果数组,结果数组中对应索引存储着下⼀个更大元素,如果没有 更大的元素,就存 -1。
比如说,输入⼀个数组 nums = [2,1,2,4,3],你返回数组 [4,2,4,-1,-1]。 解释:第⼀个 2 后面比2 大的数是 4; 1 后面比1大的数是 2;第二个 2 后面比 2 大的数是 4; 4 后⾯没有比4大的数,填 -1;3 后面没有比3 大的数,填 -1。
分析
- 这道题,我们初看,想到了使用暴力解法,直接遍历当前元素的后面的所有元素,找到第一个大于当前元素的元素。但是这样做的复杂度就直接拉满了(O(n^2))
- 我们可以把这道题给抽象:把数组的元素想象成排成一排的人,元素的大小是人的身高,我们从右往左看,抽象出来就是第一个能够看到该元素的人(身高不够的人,会被前面的人挡住)
代码
vector<int> nextGreaterElement(vector<int>& nums) {
vector<int> res(nums.size()); // 存放答案的数组
stack<int> s;
// 倒着往栈⾥放
for (int i = nums.size() - 1; i >= 0; i--) {
// 判定个⼦⾼矮
while (!s.empty() && s.top() <= nums[i]) {
// 矮个起开,反正也被挡着了。。。
s.pop();
}
// nums[i] 身后的 next great number
res[i] = s.empty() ? -1 : s.top();
s.push(nums[i]);
}
return res;
}
问题变形
739. 每日温度
分析
- 这道题与我们所给的模板大同小异,只是要求的是两个树之间的距离
- 我们稍微改变下代码即可
代码
/**
* @param {number[]} temperatures
* @return {number[]}
*/
var dailyTemperatures = function(temperatures) {
let res = [], stack = [];
for(let i = temperatures.length-1; i >= 0; i--) {
while(!(stack.length===0) && temperatures[i] >= temperatures[stack[0]]) {
stack.shift();
}
res[i] = stack.length===0 ? 0 : (stack[0]-i);
stack.unshift(i);
}
return res;
};
如何处理环形数组
分析
- 这道题又多了一个条件,那就是数组变成了环形
- 对于环形数组的处理,大家应该不会陌生了,使用除余的方法
- 其实对于这题,也是和上面使用一样的处理方式,不同的点就是如何处理最后一个数
- 最后一个数的处理必定要从第一个数开始,那么我们直接就把这个数组再复制一遍。
代码
/**
* @param {number[]} nums
* @return {number[]}
*/
var nextGreaterElements = function(nums) {
let n = nums.length;
let res = [], stack = [];
for(let i=n*2-1; i>=0; i--) {
while(!(stack.length===0) && stack[0]<=nums[i%n]) {
stack.shift();
}
res[i%n] = stack.length===0 ? -1 : stack[0];
stack.unshift(nums[i%n]);
}
return res;
};
以上题解参考labuladong的算法秘籍