每日一练python24

题目:(下一个更大元素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)

Guess you like

Origin blog.csdn.net/qq_52669357/article/details/121686092