力扣496. 下一个更大元素 I

原题

 1 class Solution:
 2     def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
 3         dic = {}
 4         ans = []
 5         lens = len(nums2)
 6         for i,n in enumerate(nums2):
 7             dic[n]=i
 8         for n in nums1:
 9             flag = False
10             for begin in range(dic[n]+1,lens):
11                 if nums2[begin] > n:
12                     ans.append(nums2[begin])
13                     flag = True
14                     break
15             if not flag:
16                 ans.append(-1)
17         return ans
18

 

posted @ 2021-01-25 10:42  凝视深空  阅读(83)  评论(0)  编辑  收藏  举报