代码随想录算法训练营第四十一天

LeetCode题目:

[739. 每日温度](#739. 每日温度)
[496. 下一个更大元素 I](#496. 下一个更大元素 I)
[503. 下一个更大元素 II](#503. 下一个更大元素 II)

其他:

739. 每日温度

跳转:
学习: 代码随想录公开讲解

问题:

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

思路:

单调栈,遇到递增弹出,并记录弹出元素与当前元素的索引差

复杂度:

时间复杂度: O ( n ) O(n) O(n)
空间复杂度: O ( n ) O(n) O(n)

代码:

java 复制代码

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

    }
}

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] 是如上所述的 下一个更大元素 。

思路:

一个数组元素在另一个数组中是否之后有更大元素,做一下映射即可.

复杂度:

时间复杂度: O ( n ) O(n) O(n)
空间复杂度: O ( n ) O(n) O(n)

代码:

java 复制代码

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

503. 下一个更大元素 II

跳转:
学习: 代码随想录公开讲解

问题:

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

数字 x 的 下一个更大的元素 是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1 。

思路:

遍历两遍即可,可以扩充数组,取余,或单纯复制一遍循环.

复杂度:

时间复杂度: O ( n ) O(n) O(n)
空间复杂度: O ( n ) O(n) O(n)

代码:

java 复制代码

class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int length = nums.length;
        int[] ans = new int[length];
        Arrays.fill(ans,-1);
        Deque<Integer> stack = new LinkedList<>();
        for(int i = 0 ; i< length ;i++){
            while(!stack.isEmpty() && nums[stack.peek()] < nums[i]){
                int push = stack.pop();
                ans[push] = nums[i];
            }
            stack.push(i);
        }
        for(int i = 0 ; i< length ;i++){
            while(!stack.isEmpty() && nums[stack.peek()] < nums[i]){
                int push = stack.pop();
                ans[push] = nums[i];
            }
            stack.push(i);
        }
        return ans;
    }
}