题目:(下一个更大元素1)给你两个 没有重复元素 的数组 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 。
程序说明:
1、暴力解题:利用index函数找出nums1中元素在nums2中的位置,由此寻找此位置的下一个位置,比较两个位置元素的大小,若符合条件则用append函数将元素添加进入新列表res
全部代码:
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
m = len(nums1)
n = len(nums2)
res = []
for i in range(m):
j = nums2.index(nums1[i])
k = j + 1
while k < n and nums2[k] < nums2[j]:
k += 1
if k < n :
res.append(nums2[k])
else :
res.append(-1)
return res
题目来源:力扣(LeetCode)