什么是单调栈?
单调栈,栈内顺序要么从大到小 要么从小到大。
一:739. 每日温度
解题思路:
遍历每日温度,维护一个单调栈,若栈为空或者当日温度小于、等于栈顶温度,则直接入栈;反之若当日温度大于栈顶温度,说明栈顶元素的升温日已经找到了,则将栈顶元素出栈,计算其与当日相差的天数即可。
注意:题目要求返回的是升温的天数,而不是升温的温度,因此栈中保存的应是数组的下标,而非温度。
代码:
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int[] result = new int[temperatures.length];
Stack<Integer> stack = new Stack();
for(int i = 0;i<temperatures.length;i++) {
while (!stack.isEmpty() && temperatures[i]>temperatures[stack.peek()]) {
// 栈里面保存的下标对应的温度是从大到小排序的
int index = stack.pop();
result[index] = i - index;
}
stack.push(i);
}
return result;
}
}
二:496. 下一个更大元素 I
解题思路:
利用 Stack、HashMap解决:
1、先遍历大数组nums2,首先将第一个元素入栈;
2、继续遍历,当当前元素小于栈顶元素时,继续将它入栈;当当前元素大于栈顶元素时,栈顶元素出栈,此时应将该出栈的元素与当前元素形成key-value键值对,存入HashMap中;
3、当遍历完nums2后,得到nums2中元素所对应的下一个更大元素的hash表;
4、遍历nums1的元素在hashMap中去查找‘下一个更大元素’,当找不到时则为-1。
代码:
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
int[] res = new int[nums1.length];
Stack<Integer> stack = new Stack();
Map<Integer,Integer> map = new HashMap();
for (int num:nums2) {
while(!stack.isEmpty() && num>stack.peek()) {
map.put(stack.pop(),num);
}
stack.push(num);
}
for(int i = 0; i<nums1.length; i++){
res[i] = map.getOrDefault(nums1[i],-1);
}
return res;
}
}