定义:
单调栈是一种比较特殊的栈,栈内元素具有单调性,单调递增或者单调递减。
单调增栈:栈顶到栈低,元素递增。
单调减栈:栈顶到栈低,元素递减。
单调增栈:
假设进栈元素为s,从栈顶开始遍历元素,把小于或等于s的元素出栈,直到出现大于s的元素或者栈为空,然后把s入栈。
伪代码:
int[] arr = new int[10];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < arr.length; i++) {
int value = arr[i];
while (!stack.isEmpty() && stack.peek() < value) {
// do something
stack.pop();
}
stack.push(i);
}
适用场景:寻找列表中右边第一个大于当前元素的值。
单调减栈:
假设进栈元素为s,从栈顶开始遍历元素,把大于或等于s的元素出栈,直到出现小于s的元素或者栈为空,然后把s入栈。
伪代码:
int[] arr = new int[10];
Stack<Integer> stack = new Stack<>();
for (int i = 0; i < arr.length; i++) {
int value = arr[i];
while (!stack.isEmpty() && stack.peek() > value) {
// do something
stack.pop();
}
stack.push(i);
}
适用场景:寻找列表中右边第一个小于当前元素的值。
举例:
现有一序列[1, 3, 2, 5, 4],求每个位置之后第一个大于当前位置的数字,按照索引输出对应序列。
构建单调增栈,入栈过程:
操作 | 栈 | 结果 |
栈空,1入栈 | [1] | |
3>1,1出栈,3入栈 | [3] | 由于3的入栈导致1被出栈,所以1之后第一个1的数一定是3,res[0]=3 |
2<3,2入栈 | [3,2] | |
5>2,2出栈;5>3,3出栈,5入栈 | [5] | res[1]=5, res[2]=5 |
4<5,4入栈 | [5,4] | 栈内还剩4和5,说明之后没有比4和5大的数,res[3]=null,res[4]=null |
所以输出的结果应该是 [3, 5, 5, null, null].
相关练习题: