例006-下一个更大的数

例006 下一个更大的数

1.问题描述

两个不重复的数组nums1和nums2，其中nums1是nums2的子集。
在nums2的相应位置找到nums1所有元素的下一个更大数字。
nums1中的数子x的下一个更大数字是nums2中x右边第一个更大的数字。
如果它不存在，则为此数字输出-1。
nums1和nums2中的所有数字都是唯一的，nums1和nums2的长度补偿过1000

2.问题示例

输入： nums1 = [4,1,2], nums2 = [1,3,4,2]
输出： [-1,3,-1]
对于第1个数组中的数字4，在第2个数组中找不到下一个更大的数字，因此输出-1
对于第1个数组中的数字1，在第2个数组中的下一个更大的数字是3
对于第1个数组中的数字2，在第2个数组中没有下一个更大的数字，因此输出-1

3.代码实现

def nextGreaterElement(nums1,nums2):
    answer = {}
    stack = []
    for x in nums2:
        while stack and stack[-1] < x:
            answer[stack[-1]] = x
            del stack[-1]
        stack.append(x)
    for x in stack:
        answer[x] = -1

    return [answer[x] for x in nums1]

nums1 = [4,1,2,12,16]
nums2 = [1,3,4,2,5,7,12,11,16,13]


print('输入： nums1 = ',nums1,'  , nums2 = ',nums2)
print('输出：',nextGreaterElement(nums1, nums2))



输入： nums1 =  [4, 1, 2, 12, 16]   , nums2 =  [1, 3, 4, 2, 5, 7, 12, 11, 16, 13]
输出： [5, 3, 5, 16, -1]