0
点赞
收藏
分享

微信扫一扫

力扣题库刷题笔记496-下一个更大元素

醉东枫 2023-04-29 阅读 65

1、题目如下:

2、个人Python代码实现

 

代码如下:

class Solution:

    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:

        #空列表用于输出结果

        ans = []

        for i in nums1:

            #如果nums2中不包含或者最后一位元素为当前遍历得元素,返回-1

            if nums2.count(i) == 0 or nums2[-1] == i:

                ans.append(-1)

            else:

                #初始化num = -1

                num = -1

                #当前元素切片后得数组进行遍历

                for j in nums2[nums2.index(i) + 1:]:

                    #如果切片后得nums2存在元素大于当前元素,则赋值给num,且退出本次循环

                    if j > i:

                        num = j

                        break

                #如果切片后的数组不存在元素大于当前元素,则num=-1

                ans.append(num)

        return ans

 

举报

相关推荐

0 条评论