Day 58|739. 每日温度| 496.下一个更大元素

最新推荐文章于 2024-09-06 10:38:06 发布

vd_vd

最新推荐文章于 2024-09-06 10:38:06 发布

阅读量85

点赞数

文章标签：算法 java 数据结构

本文链接：https://blog.csdn.net/m0_74107946/article/details/131829631

版权

这两段代码分别展示了如何利用栈数据结构解决两个数组相关问题：1)计算每日温度中比当前日更高温度的天数；2)查找数组中的下一个更大元素。第一段代码中，栈用于存储温度数组的索引，以便回溯计算等待升高天数。第二段代码中，栈用于跟踪遍历第二个数组时遇到的更大元素，同时使用HashMap快速访问第一个数组的索引，以填充结果数组。

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

● 739. 每日温度

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

● 496.下一个更大元素 I

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];
    Stack<Integer> stack = new Stack<>();
    Arrays.fill(res,-1);

    for(int i = 0;i<nums2.length;i++){
        while(!stack.isEmpty() && nums2[stack.peek()] < nums2[i]){
            int pre = nums2[stack.pop()];
            if(map.containsKey(pre)){
                res[map.get(pre)] = nums2[i];
            }
        }
        
        stack.push(i);
    }
    return res;
    }
}