18.四数之和--python

题:给定一个包含n个整数的数组nums和一个目标值target,判断nums中是否存在四个元素,使他们的和为target?

法:转换为两数之和做,建立一个字典,key值是两数之和,value是他们成对的下标,一个key可能对应多组下标,所以value是列表的形式,表中每个元素都是一个下标对。建立完字典后,再去遍历用两个指针i与j暴力遍历nums,每次都去找tmp=target-nums[i]-nums[j],如果tmp在字典中,说明对于下标i与j,还有另外的两个数与他们相加等于target,找到dict1[key]对应的下标对即可。这里注意,为了避免出现重复四元组,我们先将nums排序,并对于i与j下标,判断找到的新下标是否大于j

def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        length=len(nums)
        if length<4:return []
        
        dict1={}
        nums.sort()
        for i in range (length):
            for j in range (i+1,length):
                tmp=nums[i]+nums[j]
                if tmp not in dict1:dict1[tmp]=[(i,j)]
                else:
                    dict1[tmp].append((i,j))
        
        
        res=set()
        for i in range (length-3):
            for j in range (i+1,length-2):
                tmp=target-nums[i]-nums[j]
                if tmp in dict1:
                    for k in dict1[tmp]:
                        if k[0]>j:
                            res.add((nums[i],nums[j],nums[k[0]],nums[k[1]]))
        #print(res)
        return list(res)

 

 

 

 

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值