python 单调栈

496. 下一个更大元素 I

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。

给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

这是leetcode中的一道题，可以通过维护一个单调递增的单调栈来帮助解决，下面给出代码和解释

# 示例：nums1 = [4,1,2] nums2 = [1,3,4,2] 输出 [-1,3,-1] 
class Solution:
    def nextGreaterElement(self, nums1: list[int], nums2: list[int]) -> list[int]:
        n1 = len(nums1)
        n2 = len(nums2)
        # 首先找到num2中每个数字右侧比他大的第一个数 若找不到，对应位置返回-1
        d = {}
        stack = [] # 需要维护的单调不减的栈
        res = [-1] * n1 # 初始化最后需要返回的数组
        for i in range(n2):
            d[nums2[i]] = -1   # 初始化哈希表,后续通过哈希表来查找时间复杂度为O(1)
        for k in range(n2):
            num = nums2[k]
            while stack and stack[-1] < num:   # 维护单调不减的栈
                x = stack.pop()
                d[x] = num
            stack.append(num)
        for i in range(n1):
            res[i] = d[nums1[i]]
        return res
if __name__ == '__main__':
    s = Solution()
    print(s.nextGreaterElement([4,1,2],[1,3,4,2])) # [-1,3,-1]
# 总体时间复杂度应该为O(n1+n2)