70-Stack--LC496下一个更大元素I

原创已于 2022-03-17 01:27:32 修改 · 105 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

于 2022-03-17 01:26:50 首次发布

leetcode 专栏收录该内容

75 篇文章

订阅专栏

本文介绍了如何使用暴力法和两种单调栈策略（后往前和往前往后）解决LeetCode中的下一个更大元素问题。通过实例代码展示了如何在不同场景下找到nums1数组中每个元素的下一个大于它的元素，以及哈希表在优化搜索过程中的作用。

在这里插入图片描述

三种方法

class Solution(object):
    def nextGreaterElement(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        # 1.先来个暴力法
        res = [0 for i in range(len(nums1))]
        for i in range(len(nums1)):
            j = nums2.index(nums1[i])
            k = j + 1
            while k < len(nums2) and nums2[k] <= nums1[i]:
                k += 1
            res[i] = nums2[k] if k < len(nums2) else -1
                
                
        # return res
        # 2.单调栈：一但要求下一个更大的元素，就是用单调栈解
        # 单调栈分为两种，一种从后往前遍历，一种从前往后遍历
        # 2.1 从后往前
        res = {}  # 用哈希表存储
        stack = []
        for num in reversed(nums2):
            while stack and num > stack[-1]:
                stack.pop()
            res[num] = stack[-1] if stack else -1
            stack.append(num)
        return [res[num] for num in nums1]
        # 2.2 从前往后
        res = {}
        stack = []
        for num in nums2:
            while stack and num > stack[-1]:
                temp = stack.pop()
                res[temp] = num
            
            stack.append(num)
        # 循环结束，如果stack中还有元素，则为单调递增的，没有下一个最大元素
        # 2.1则少了这一步
        if stack:
            for i in stack:
                res[i] = -1
        return [res[num] for num in nums1]