58-单调栈-739. 每日温度 496.下一个更大元素 I

文章目录

  • 739. 每日温度
  • 496.下一个更大元素 I

739. 每日温度

请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。

注意:暴力解法会超出时间限制。

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        /*
        int[] res = new int[temperatures.length];
        for(int i = 0; i < temperatures.length; i ++){
            for(int j = i + 1; j < temperatures.length; j ++){
                if(temperatures[j] > temperatures[i]){
                    res[i] = j - i;
                    break;
                }
            }
        }
        return res;*/
        int len = temperatures.length;
        int[] res = new int[len];
        
        Deque<Integer> stack = new LinkedList<>();
        stack.push(0);
        for(int i = 1; i < len; i ++){
            while(!stack.isEmpty() && temperatures[i] > temperatures[stack.peek()]){
                res[stack.peek()] = i - stack.peek();
                stack.pop();
            }
            stack.push(i);
        }
        return res;
    }
}

496.下一个更大元素 I

给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
示例 1:输入: nums1 = [4,1,2], nums2 = [1,3,4,2].输出: [-1,3,-1]

思路:跟上一个相似,因为要在第二个数组中选择元素,所以仍然是遍历第二个数组,不同的在于采用map存储第一个数组从数值到下表的对应关系。
注意:1.containsKey 2.注意pop的位置。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int len = nums1.length;
        int[] res = new int[len];
        Arrays.fill(res, -1);
        
        HashMap<Integer, Integer> map = new HashMap<>();
        for(int i = 0; i < len; i ++){
            map.put(nums1[i], i);
        }
        
        Deque<Integer> stack = new LinkedList<>();
        stack.push(nums2[0]);
        for(int i = 1; i < nums2.length; i ++){
            while(!stack.isEmpty() && nums2[i] > stack.peek()){
                if(map.containsKey(stack.peek())){
                    res[map.get(stack.peek())] = nums2[i];
                }
                stack.pop();
            }
            stack.push(nums2[i]);
        }
        return res;
    }
}

你可能感兴趣的:(LeetCode,算法,数据结构,leetcode)