代码随想录算法训练营-day58-739. 每日温度、496.下一个更大元素 I

最新推荐文章于 2024-09-14 22:25:23 发布

依旧1919

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量156

点赞数

分类专栏：代码随想录算法训练营文章标签：栈每日温度下一个更大元素编程题数据结构

本文链接：https://blog.csdn.net/weixin_47067912/article/details/128432018

版权

代码随想录算法训练营专栏收录该内容

53 篇文章 0 订阅

订阅专栏

739. 每日温度

学习文章链接：
思路：当出现后面值大于前面的时候，就开始给结果数组赋值。
代码：

class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        Stack<Integer> st = new Stack<>();
        int[] res = new int[temperatures.length];
        st.push(0);
        for (int i = 1; i < temperatures.length; i++) {
            if (temperatures[i] < temperatures[st.peek()]) {
                st.push(i);
            } else if (temperatures[i] == temperatures[st.peek()]) {
                st.push(i);
            } else {
                while (!st.isEmpty() && temperatures[i] > temperatures[st.peek()]) {
                    res[st.peek()] = i - st.peek();
                    st.pop();
                }
                st.push(i);
            }
        }
        return res;
    }
}

496.下一个更大元素 I

学习文章链接：
思路：见代码
代码：

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