496. 下一个更大元素 I.py 645 B

1234567891011121314151617181920212223
  1. from typing import List
  2. class Solution:
  3. def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
  4. res = []
  5. num2_dict = {}
  6. for i in range(len(nums2)):
  7. num2_dict[nums2[i]] = i
  8. for x in nums1:
  9. exist = False
  10. for j in range(num2_dict[x], len(nums2)):
  11. if nums2[j] > x:
  12. res.append(nums2[j])
  13. exist = True
  14. break
  15. if not exist:
  16. res.append(-1)
  17. return res
  18. if __name__ == '__main__':
  19. print(Solution().nextGreaterElement([4, 1, 2], [1, 3, 4, 2]))