2021 10/26：记住，一但要求下一个更大的元素，就是用单调栈解(stack)

最新推荐文章于 2024-10-30 13:16:11 发布

时俗之俗

最新推荐文章于 2024-10-30 13:16:11 发布

阅读量129

点赞数

分类专栏：日记文章标签： python

本文链接：https://blog.csdn.net/weixin_40857308/article/details/120983079

版权

日记专栏收录该内容

24 篇文章 0 订阅

订阅专栏

下一个更大元素 I

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        res={}
        stack=[]
        for i in range(len(nums2)-1,-1,-1):
            while stack and stack[-1]<=nums2[i]:
                stack.pop()
            res[nums2[i]] = stack[-1] if stack else -1
            stack.append(nums2[i])
        return [res[num] for num in nums1]

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        Deque<Integer> stack = new ArrayDeque<Integer>();
        for (int i = nums2.length - 1; i >= 0; --i) {
            int num = nums2[i];
            while (!stack.isEmpty() && num >= stack.peek()) {
                stack.pop();
            }
            map.put(num, stack.isEmpty() ? -1 : stack.peek());
            stack.push(num);
        }
        int[] res = new int[nums1.length];
        for (int i = 0; i < nums1.length; ++i) {
            res[i] = map.get(nums1[i]);
        }
        return res;
    }
}\