496、下一个更大元素

最新推荐文章于 2024-09-21 10:02:17 发布

代码与思路齐飞

最新推荐文章于 2024-09-21 10:02:17 发布

阅读量2.8k

点赞数

分类专栏：算法刷题 python 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/dsl200970/article/details/120969497

版权

算法刷题同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

python

7 篇文章 0 订阅

订阅专栏

解题思路

既然要找元素后面更大的元素，如果从nums1遍历找，我们不知道nums1中元素在nums2中的位置，所以直接从nums2遍历查找，另外需要一个临时列表存放未找到更大值的元素以及一个字典存放元素及其对应最大值。如果当前列表中值小于遍历值，将其与遍历值存入字典，然后移除，不断循环。否则将遍历值存入列表，进行下一次比较。

最后遍历nums1，通过字典找到其对应的最大值。

代码

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        stack, hashmap = list(), dict()
        for num in nums2:
            while stack and stack[-1] < num:
                hashmap[stack.pop()] = num
            stack.append(num)
        return [hashmap.get(i, -1) for i in nums1]

链接：https://leetcode-cn.com/problems/next-greater-element-i/solution/496-xia-yi-ge-geng-da-yuan-su-by-senlind-6fxp/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。