1、问题描述
两个不重复元素的数组 nums1 和 nums2,其中nums1 是 nums2 的子集。在nums2相应位置找到nums1所有元素的下一个更大数字。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
nums1 和 nums2中所有数字都是唯一的, nums1 和 nums2的长度不超过1000。
2、问题示例
输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。
对于 num1 中的数字 1 ,第二个数组中数字1右边的下一个较大数字是 3 。
对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1 。
3、代码实现
class Solution:
def nextGreaterElement(self, nums1, nums2):
answer = {}
stark = []
for x in nums2:
while stark and stark[-1] < x:
answer[stark[-1]] = x
del stark[-1]
stark.append(x)
for x in stark:
answer[x] = -1
return[answer[x] for x in nums1]
if __name__ == '__main__':
s = Solution()
nums1 = [4,1,2]
nums2 = [1,3,4,2]
print("输入1:",nums1)
print("输入2:",nums2)
print("输出:", s.nextGreaterElement(nums1,nums2))