前言
单调栈在算法中,能够将O(n^2)的时间复杂度优化到O(n),相对的空间复杂度会增加,因为需要动态维护一个栈。在此总结一下 单调栈 题目,题目来源:
496. 下一个更大元素 I
503. 下一个更大元素 II
一、496. 下一个更大元素 I题目总结
496. 下一个更大元素 I
该方法核心是寻找到数组nums2各个元素的下一个更大元素,其实现方法为使用单调栈方法,又分为逆序寻找和顺序寻找,在此总结一下:
1.1 逆序寻找
其核心思想是使得栈中的元素满足单调递增,当不满足条件时进行弹出,以保持此特性。个人理解为:对于数组某一元素num,我们将其与栈顶的元素peekNum进行比较:
细节说明:
维持栈的单调递增特性本质上是 对元素num右侧的所有元素在不打乱原有的顺序下选出升序排序序列。 例如数组{8,2,3,4,1,10},对于元素8,其右侧元素的升序排序序列为:2,3,4,10, 可以发现此序列中10比8大,故8所对应的下一个更大的元素为10;
代码实现:
// 逆序寻找下一个更大的数字
for (int i = nums2。length-1; i >=0; i--){
int num = nums2[i];
while (!stack.isEmpty() && num > stack.peek()){
stack.pop();
}
map.put(num, stack.isEmpty() ? -1 : stack.peek());
stack.push(num);
}
1.2 顺序寻找
其核心思想是令数组中的元素依次入栈,当遍历到某个元素num,大于栈顶的元素peekNum时,说明此时peekNum的下一个更大元素为num,记录下即可。
代码实现:
// 正序寻找下一个更大的数字
for (int i = 0; i < nums2.length; i++){
int num = nums2[i];
while (! stack.isEmpty() && num > stack.peek()){
map.put(stack.pop(), num);
}
stack.push(num);
}ck.push(num);
}
二、503. 下一个更大元素 II
503. 下一个更大元素 II
其实现方法与 1.2顺序寻找方法一致,唯一区别是要遍历数组两遍,因为不只是寻找某元素右侧的下一个更大元素,而是按照顺时针循环寻找下一个更大的元素。
代码实现:
class Solution {
public int[] nextGreaterElements(int[] nums) {
int n = nums.length;
int[] res = new int[n];
Arrays.fill(res, -1);
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < 2*n; i++){
while (!stack.isEmpty() && nums[ i % n] > nums[stack.peek() % n]){
int index = stack.pop();
res[index] = nums[i%n];
}
stack.push(i%n);
}
return res;
}
}