刷题第57天 | 739. 每日温度、496.下一个更大元素 I

739. Daily Temperatures

题目链接:739. Daily Temperatures
思路链接:代码随想录单调栈-每日温度

思路

需要理解单调栈的原理。首先将第0个下标push进入栈中,然后从1开始遍历数组。如果在数组中遍历到的元素小于等于单调栈顶的下标对应的元素,也就是之前遍历过的元素,就直接将对应下标push入栈。如果在数组中遍历到的元素大于单调栈顶的下标对应的元素(在数组中,在该元素右侧,找到了大于该元素的元素),就需要在result数组相应的位置记录下来,并且在栈中pop掉该元素下标,因为已经找到了。然后再将当前遍历的元素的下标push入栈,因为还没有找过大于当前元素的右侧的值。

Code

import java.util.Stack;
class Solution {
    public int[] dailyTemperatures(int[] temperatures) {
        // 递增栈
        Stack<Integer> st = new Stack();
        int[] result = 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 {
                while (!st.empty() && temperatures[i] > temperatures[st.peek()]) {
                    result[st.peek()] = i - st.peek();
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
}

496. Next Greater Element I

题目链接:496. Next Greater Element I
思路链接:代码随想录单调栈-下一个更大元素 I

思路

通过map来映射nums1中元素与元素对应的下标,然后利用单调栈,遍历nums2,在nums2中,找出第一个大于nums1中元素的元素,然后记录下来。

Code

import java.util.Map;
import java.util.HashMap;
import java.util.Stack;
import java.util.Arrays;

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int[] result = new int[nums1.length];
        Arrays.fill(result, -1);
        // 递增栈
        Stack<Integer> st = new Stack<>();
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            map.put(nums1[i], i); // key: element; value: index
        }
        st.push(0);
        for (int i = 1; i < nums2.length; i++) {
            if (nums2[i] <= nums2[st.peek()]) {
                st.push(i);
            } else {
                while (!st.empty() && nums2[i] > nums2[st.peek()]) {
                    if (map.containsKey(nums2[st.peek()])) {
                        result[map.get(nums2[st.peek()])] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
}

你可能感兴趣的:(leetcode,算法,java,数据结构,单调栈)