LeetCode 496. 下一个更大元素 I

496. 下一个更大元素 I

 

【单调栈】从后往前遍历，对于元素i，不断弹出比i小的元素，如果栈不为空，那么栈顶就是第一个比i大的元素，然后把i押入栈。

class Solution {

    // 单调栈找右侧第一个大于他的元素
    // 1:55 2:14

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Deque<Integer> stack = new LinkedList();
        Map<Integer, Integer> map = new HashMap();
        for (var i = 0; i < nums2.length; i++) {
            map.put(nums2[i], i);
        }
        int[] arr = new int[nums2.length];
        for (var i = nums2.length - 1; i >= 0; i--) {
            while (!stack.isEmpty()) {
                int top = stack.peek();
                if (top > nums2[i]) {
                    arr[i] = top;
                    stack.push(nums2[i]);
                    break;
                } else {
                    stack.poll();
                }
            }
            if (stack.isEmpty()) {
                arr[i] = -1;
                stack.push(nums2[i]);
            }
        }
        int[] ans = new int[nums1.length];
        for (var i = 0; i < nums1.length; i++) {
            ans[i] = arr[map.get(nums1[i])];
        }
        return ans;
    }
}