单调栈(栈中的元素单调递增或者单调递减)----主要解决“The Next Greater Number”问题,寻找下一个更大或者下一个更小的数。
(以单调递增为例)实现的思路就是:从后往前遍历,如果当前元素比栈顶元素大,则栈顶元素出栈,再接着判断,直到栈为空或者栈内的数大于当前元素为止;如果当前元素比栈顶元素小,则直接进栈。具体题目及代码如下:
给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Map<Integer, Integer> map = new HashMap<Integer, Integer>(); //hash用来记录比当前元素大的下一个元素,通过key,value的方式将二者绑定。
Deque<Integer> stack = new ArrayDeque<Integer>();
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()); //此时的栈顶元素就是第一个比当前元素大的数,将两者组合成(key,value)的形式保存起来
stack.push(num); //比栈顶小,直接进栈。
}
int[] res = new int[nums1.length];
for (int i = 0; i < nums1.length; ++i) {
res[i] = map.get(nums1[i]);
}
return res;
}
}
基本原理:
倒序保证了,只有后面的数比前面的数大才有机会保存在栈里。如果比前面的数小,那么在找The Next Greater Number时,是不可能轮到它的。所以每次进栈的操作,实际上就是在清除当前元素和下一个较大元素之间的较小的元素,就像在记录“山头”一样。