【leetcode - 496】下一个更大元素 -Easy

给定两个没有重复元素的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出-1。

示例 1:

输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
对于num1中的数字4，你无法在第二个数组中找到下一个更大的数字，因此输出 -1。
对于num1中的数字1，第二个数组中数字1右边的下一个较大数字是 3。
对于num1中的数字2，第二个数组中没有下一个更大的数字，因此输出 -1。
示例 2:

输入: nums1 = [2,4], nums2 = [1,2,3,4].
输出: [3,-1]
解释:
  对于num1中的数字2，第二个数组中的下一个较大数字是3。
对于num1中的数字4，第二个数组中没有下一个更大的数字，因此输出 -1。
注意:

nums1和nums2中所有元素是唯一的。
nums1和nums2 的数组大小都不超过1000。

【original】

思路： 考虑nums1中每个元素在num2中位置，选择切片后面的位置，找到第一个大于这个元素的值，跳出循环

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        dic = {}
        stack = []
        for i, j in enumerate(nums2):
            dic[j] = -1
            for k in nums2[i+1:]:
                if k>j:
                    dic[j] = k
                    break
        for p in nums1:
            if p in nums2:
                stack.append(dic[p])
            else:
                stack.append(-1)
        return stack

original

【upgrade】

思路： 维护一个单调栈，在遍历nums2时，如果栈不为空，且新的元素大于上一个元素，则新元素作为现有栈中所有元素的下一个更大元素，即，判断新元素大于栈中最后一个元素，结果为正则出栈，不断循环，直到为空，每一次将字典添加新的值。若不大于栈内最后一个元素，入栈。最后从字典中取nums1值对应的结果，如果不存在就返回-1

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        dic = {}
        stack = []
        for i in nums2:
            while stack and stack[-1]<i:
                dic[stack.pop()] = i
            stack.append(i)
        return [dic.get(i,-1) for i in nums1]

Upgrade