Leetcode 496. Next Greater Element I

最新推荐文章于 2025-12-08 22:55:11 发布

原创最新推荐文章于 2025-12-08 22:55:11 发布 · 201 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Leetcode 专栏收录该内容

409 篇文章

订阅专栏

博主Tyan分享了两种方法解决LeetCode问题Next Greater Element I，Version1利用双重循环查找并存储结果，Version2则借助栈优化搜索。适合理解不同查找策略在编程中的应用。

文章作者：Tyan
博客：noahsnail.com | 优快云 | 简书

1. Description

Next Greater Element I

2. Solution

**解析：**Version 1，由于元素是唯一的，通过循环找出每个nums2中的满足条件结果保存到字典中，遍历nums1，获得结果。Version 2通过使用栈来寻找满足条件的结果，减少搜索时间。

Version 1

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        stat = {}
        for i in range(len(nums2)):
            for j in range(i+1, len(nums2)):
                if nums2[j] > nums2[i]:
                    stat[nums2[i]] = nums2[j]
                    break
            if nums2[i] not in stat:
                stat[nums2[i]] = -1
        result = [stat[x] for x in nums1]
        return result

Version 2

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        stat = {num: -1 for num in nums2}
        stack = []
        for num in nums2:
            while stack and stack[-1] < num:
                stat[stack.pop()] = num
            stack.append(num)
        result = [stat[x] for x in nums1]
        return result