LeetCode0496. 下一个更大元素 I

一. 题目

1. 题目
   

2. 示例
   

二. 方法一: 单调栈 + map

1. 解题思路

   1. 用单调栈查找nums2中每个元素右侧比他大的第一个元素
   2. 再用map存储nums2中当前元素和他下一个元素的值
   3. 再遍历nums1中的元素:判断是否在map中,:
      若在, 则返回其对应的value
      若不在, 则返回-1

2. 解题代码

   def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
       stack = []
       map1 = {}
       result = []
       for index in range(len(nums2)):
           while stack and nums2[index] > nums2[stack[-1]]:
               map1[nums2[stack[-1]]] = nums2[index]
               stack.pop()
           stack.append(index)
       for ele in nums1:
           if ele in map1.keys():
               result.append(map1[ele])
           else:
               result.append(-1)
   
       return result
   

3. 分析
   时间复杂度: O(n + m) 其中n为nums2中元素的数量, m为nums1中元素的数量
   空间复杂度: O(n)

三. 方法二

1. 解题思路

2. 解题代码

3. 分析