刷题_leetcode_496. 下一个更大元素 I

496. 下一个更大元素 I - 力扣（LeetCode）

该题我最开始的想法是直接暴力求解，但是转念一想这样的话，时间复杂度可能会达到O(m*n)。于是就采用了单调栈+哈希值。

单调栈的思路：从栈底到栈顶的元素是单调递减的，通过遍历逆置的数组，来实现找到右边第一个更大的元素。当遍历到当前元素时，将其与栈顶比较，小则直接入栈，大则不断弹出，直到找到合适自己的位置或者栈空。并且在每次找到合适位置之后，插入之前，更新哈希表。

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