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

739. 每日温度

题目链接：力扣

题目要求：

        给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

示例 1:

输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]

• 1 <= temperatures.length <= 105
• 30 <= temperatures[i] <= 100

总结：

        一般遇到求一个位置之后或之前的大于它或小于它的这种问题，都是单调栈问题，我们用result数组记录最终我们需要返回的值，result[i]代表第i天的温度在其后几天后能找到一个高于它的温度，创建一个栈，用来存储遍历天数的下标，先把第0天的下标入栈，之后依次遍历每天，如果遍历的i天的温度小于或者等于栈顶的天数的温度，则把i天的i存入栈，如果栈不为空，且遍历的该天大于栈顶的天数的温度，则说明栈顶那天的温度的result值找到了，我们只需要把这两天的天数作差，即可得到栈顶那天的result值，之后把栈顶那天的记录弹出，再用遍历到的i天的温度与新的栈顶那天的温度比较，重复上述操作，一直到不大于栈顶那天的温度或者栈里没元素了，则将该天加入栈中，最后每天都遍历结束，则返回result即可。

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

496. 下一个更大元素 I

题目链接：力扣

题目要求：

  nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

示例 1：

输入：nums1 = [4,1,2], nums2 = [1,3,4,2].
输出：[-1,3,-1]
解释：nums1 中每个值的下一个更大元素如下所述：
- 4 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。
- 1 ，用加粗斜体标识，nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ，用加粗斜体标识，nums2 = [1,3,4,2]。不存在下一个更大元素，所以答案是 -1 。

• 1 <= nums1.length <= nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 104
• nums1和nums2中所有整数 互不相同
• nums1 中的所有整数同样出现在 nums2 中

总结：

        这里结果数组result的大小应该是nums1的长度，因为是说nums1中的每个对应元素下一个比他大的元素是多少，这里我们的想法是先用map记录nums1中的元素值和其对应的下标值，使其一一对应，我们对于nums2还用上题类似的方法，如果遍历到一个位置的元素大于栈顶存的那个位置的元素，则说明栈顶存的那个位置的元素找到了其下一个更大的元素，则看栈顶存的那个位置的元素值，在map中也就是nums1中是否有该元素，如果有，则需要将该元素在在map中的对应位置，也就是nums1中对应位置的result数组赋值为遍历的大于栈顶位置元素的那个位置的元素值，其余操作与上一题一致，最终返回result即可。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack stack = new Stack<>();
        int[] result = new int[nums1.length];
        Arrays.fill(result,-1);
        HashMap hashmap = new HashMap<>();
        for(int i = 0;i < nums1.length;i++){
            hashmap.put(nums1[i],i);
        }
        stack.push(0);
        for(int i = 1;i < nums2.length;i++){
            if(nums2[i] <= nums2[stack.peek()]){
                stack.add(i);
            }else{
                while(!stack.isEmpty() && nums2[i] > nums2[stack.peek()]){
                    //如果找到了栈顶的元素的下一个更大元素，看栈顶元素在nums1中是否存在，存在则收获结果
                    if(hashmap.containsKey(nums2[stack.peek()])){
                        result[hashmap.get(nums2[stack.peek()])] = nums2[i];
                    }
                    stack.pop();
                }
                stack.add(i);
            }
        }
        return result;
    }
}