代码随想录训练营第58天|739.每日温度，496.下一个更大元素I

739.每日温度

文章

代码随想录|0739.每日温度

思路

单调栈里存放温度数组下标，维持从栈底到栈顶是温度递减的
单调栈中下标的含义是，还没有找到右侧第一个温度比他高的日期
所以如果访问到某个日期的温度比栈顶元素高则一定是第一个出现的比栈顶元素温度高的日期，此时要pop

代码

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

496.下一个更大元素I

文章

代码随想录|0496.下一个更大元素I

思路

先用HashMap做一个nums1中值到下标的映射，然后再从nums2中找每个元素的下一个更大元素，找到之后访问map填入新数组中

代码

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

总结

前两天做过每日温度了其实
主要是辨析单调栈和单调队列的区别，加深一下理解