代码随想录算法训练营第六十天 | 739. 每日温度 & 496.下一个更大元素 I

1. 每日温度 

739. 每日温度 - 力扣（LeetCode）

维护一个单调递减（可以等于）的栈，栈内存的是元素的下标

当前元素小于等于栈顶元素时，直接入栈

否则，将栈顶元素出栈（对于栈顶元素，它找到了它的答案，即下标之差）

循环比较记录结果，直到当前元素入栈后，栈依然单调递减

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int length = temperatures.length;
        int[] res = new int[length];
        Stack stack = new Stack<>();
        stack.push(0);
        for(int i = 1; i < length; i++){
            if(temperatures[i] <= temperatures[stack.peek()]){
                stack.push(i);
            }else{
                while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){
                    res[stack.peek()] = i - stack.peek();
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return res;
    }
}

2. 下一个更大元素 I 

496. 下一个更大元素 I - 力扣（LeetCode）

和前面差不多，只不过需要返回的结果要处理下

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Map map = new HashMap<>();
        for(int i = 0; i < nums1.length; i++){
            map.put(nums1[i], i);
        }

        Stack stack = new Stack<>();
        stack.push(nums2[0]);
        int[] res = new int[nums1.length];
        Arrays.fill(res, -1);
        for(int i = 0; i < nums2.length; i++){
            if(nums2[i] <= stack.peek()){
                stack.push(nums2[i]);
            }else{
                while(!stack.isEmpty() && nums2[i] > stack.peek()){
                    if(map.containsKey(stack.peek())){ // 只有当栈顶元素需要被记录时才记录
                        int idx = map.get(stack.peek()); // 取出记录该在的位置
                        res[idx] = nums2[i];
                    }
                    stack.pop(); // 不需要记录的元素也需要出栈，继续看前一个元素
                }
                stack.push(nums2[i]);
            }
        }

        return res;
    }
}