北京做网站浩森宇特网店运营推广
题目:(下一个更大元素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 + 1while k < n and nums2[k] < nums2[j]:k += 1if k < n :res.append(nums2[k]) else :res.append(-1)return res
题目来源:力扣(LeetCode)