力扣496题(单调栈)

496.下一个最大元素I

基本思想:

先对num1中每一个元素他的下一个最大元素对应的写成个字典,

再将num2中每一个元素在字典中找到对应的value放到一个列表中

具体实现:

stack是栈

stack[-1]是列表中最后一个元素

stack.pop()弹出的是最后一个元素--------出栈

stack.append()在列表末尾加入一个元素--------入栈

代码:

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        # time O(n), space O(n)
        
        stack = []
        book = {}
        for i in range(len(nums2)):
            while stack and stack[-1] < nums2[i]:
                book[stack.pop()] = nums2[i]
            stack.append(nums2[i])
        
        res = []
        
        for n in nums1:
            if n in book:
                res.append(book[n])
            else:
                res.append(-1)
        
        return res

 

上一篇:496. 下一个更大元素 I


下一篇:Java每日一题:496. 下一个更大元素 I