Problem: 503. 下一个更大元素 II
思路
单调栈走起,维护一个单调递减栈。
解题方法
牺牲一下空间复杂度,开辟一个2*n的数组,遍历此数组。这样对于前置数组的末尾那几位,就可以通过继续遍历后置数组,从而能够正确获取到最近的比它要大的数。并通过一个
max来记录数组中最大的值
举个例子:假设nums是[1,2,3,2,1]
则2*n的newNums就是[1,2,3,2,1,1,2,3,2,1]
当我们遍历到前置数组 的最后一项 ,1时

此时的单调栈如下:
[2,3,4]
注意哦,单调栈中保存的是数组下标index哦!!!
也就是说,此时的结果数组为:[2,3,待定,待定,待定,待定,待定,待定,待定,待定]
结果数组也是2*n
只遍历一次显然不够,于是继续遍历后置数组
[1,2,3,2,1]
逻辑同第一次遍历,有区别的是,在第二次遍历假如当前项和最大值相同,则要将对应的结果数组 中的index置为-1即可。
res[i - nums.length] = -1
最后我们得到这样的结果数组 :
[ 2, 3, -1, 3, 2, 2, 3, 0, 0, 0 ]
返回前n项即可:
[ 2, 3, -1, 3, 2]
复杂度
时间复杂度:
O(n)
空间复杂度:
O(n)
Code
js
/**
* @param {number[]} nums
* @return {number[]}
*/
var nextGreaterElements = function (nums) {
let stk = [];
let res = Array(nums.length * 2).fill(0);
let newNums = [...nums, ...nums]
let max = -Infinity
for (let i = 0; i < newNums.length; i++) {
while (stk.length > 0 && newNums[stk[stk.length - 1]] < newNums[i]) {
res[stk.pop()] = newNums[i]
}
if (max < newNums[i]) {
max = newNums[i]
}
if (newNums[i] === max && i >= nums.length) {
res[i - nums.length] = -1;
}
stk.push(i)
}
return res.slice(0, nums.length)
};