739. 每日温度
使用单调栈的情况:一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置。 本题其实就是找找到一个元素右边第一个比自己大的元素,此时就应该想到用单调栈了。
单调栈的本质是用空间换时间,因为在遍历的过程中需要一个栈来记录右边第一个比当前元素高的元素,优点是整个数组只需要遍历一次。总的来说就是用一个栈来记录我们遍历过的元素。这里我的理解是,当一个数组遍历到的数与栈顶相比更大时,就把原来的栈顶数弹出,把数组中这个更大的数作为栈顶。
class Solution {
public int[] dailyTemperatures(int[] temperatures) {
int lens = temperatures.length;
int[] res = new int[lens];
Deque<Integer> stack = new LinkedList<>();
for(int i = 0; i < lens; i ++){
while(!stack.isEmpty()&&temperatures[i]>temperatures[stack.peek()]){
res[stack.peek()] = i - stack.peek();
stack.pop();
}
stack.push(i);
}
return res;
}
}
496. 下一个更大元素 I
本题和739. 每日温度是一样的。栈头到栈底的顺序,要从小到大,也就是保持栈里的元素为递增顺序。只要保持递增,才能找到右边第一个比自己大的元素。
class Solution {
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
Stack<Integer> temp = new Stack<>();
int[] res = new int[nums1.length];
Arrays.fill(res,-1);
HashMap<Integer, Integer> hashMap = new HashMap<>();
for (int i = 0 ; i< nums1.length ; i++){
hashMap.put(nums1[i],i);
}
temp.add(0);
for (int i = 1; i < nums2.length; i++) {
if (nums2[i] <= nums2[temp.peek()]) {
temp.add(i);
} else {
while (!temp.isEmpty() && nums2[temp.peek()] < nums2[i]) {
if (hashMap.containsKey(nums2[temp.peek()])){
Integer index = hashMap.get(nums2[temp.peek()]);
res[index] = nums2[i];
}
temp.pop();
}
temp.add(i);
}
}
return res;
}
}