Leetcode ----- 496 下一个更大元素(python)

找下一个更大元素

给定两个没有重复元素的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出-1。

思路一:

  1. 循环两个列表,在nums2中找nums1中的数字。如果没有找到在新列表中加-1
  2. 如果找到了,判断他是否在nums2的末尾,如果是,直接在新列表中加-1
  3. 如果不是,从j+1以后找,直到找到有比nums2[j]大的数字,如果没有直接在新列表中加-1,如果有在新列表中加找到的那个值。
超出时间限制!!!!
class Solution(object):
    def nextGreaterElement(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        nums3=[]
        for i in range(len(nums1)):
            for j in range(len(nums2)):
                if nums1[i] == nums2[j]:
                    if j == len(nums2)-1:
                        nums3.append(-1)
                    else: 
                        for k in range(j+1,len(nums2)):
                            while nums1[i] < nums2[k]:
                                nums3.append(nums2[k]
                            nums3.append(-1)                          
                else:   
                    j = j + 1
        return nums3

思路二:

  1. 运用了栈的思想,以及dict
  2. 将nums2的第一个数进栈,继续遍历,比较当前的元素与栈顶相比较,如果比栈顶大,那么他们组成了一个dict{res.pop():nums2[j]},如果不是,进栈
  3. 遍历nums2以后,这样就得到了nums2中所对应的下一个最大元素
  4. 遍历nums1看dict中是否存在,如果不存在,新列表加-1,如果存在,加dict中key对应的value。
class Solution(object):
    def nextGreaterElement(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        res = []
        dict1 ={}
        nums3 = []
        for i in range(len(nums2)):
            while not res ==[] and nums2[i]>res[-1]:  
                dict1[res.pop()]= nums2[i]
            res.append(nums2[i])   
    
        for k in range(len(nums1)):
            if  dict1.has_key(nums1[k]):
                nums3.append(dict1[nums1[k]])
            else:
                nums3.append(-1)
        return nums3
                    
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值