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

最新推荐文章于 2024-07-19 15:45:46 发布

麦城稻香咖啡

最新推荐文章于 2024-07-19 15:45:46 发布

阅读量202

点赞数 2

文章标签：算法

本文链接：https://blog.csdn.net/m0_56545264/article/details/136474798

版权

文章介绍了如何运用单调栈算法来解决LeetCode中的两个问题：每天温度数组中找到每个温度后需要等待的天数，以及在一个有序数组中找到每个元素的下一个更大值。作者给出了Java实现的代码示例。

摘要由CSDN通过智能技术生成

739. 每日温度

题目链接：. - 力扣（LeetCode）

解题思路：单调栈

java：

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        int lens=temperatures.length;
        int []res=new int[lens];
        Deque<Integer> stack=new LinkedList<>();
        stack.push(0);
        for(int i=1;i<lens;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;
    }
}

496.下一个更大元素 I

题目链接：. - 力扣（LeetCode）

解题思路：还是单调栈

java：

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