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