leetcode18:四树之和

98 篇文章 1 订阅
83 篇文章 1 订阅

在这里插入图片描述
思路:leecode16是三数之和,这题可以借鉴,将四数之和转化为三数之和。

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        nums.sort()
        res = []
        for i in range(len(nums)):
            for j in range(i+1,len(nums)):  #多加一层循环
                p1 = j+1
                p2 = len(nums)-1
                while p1 < p2:
                    c = nums[i] + nums[j] + nums[p1]  + nums[p2] 
                    if c == target and [nums[i],nums[j],nums[p1],nums[p2]] not in res:  #多加一个判断条件去重
                        res.append([nums[i],nums[j],nums[p1],nums[p2]])
                        p1 += 1
                        # p2 -= 1
                    elif c < target:
                        p1 += 1
                    else:
                        p2 -= 1
        return res

注:不加判断条件的话,返回也可写成 return list(set(map(tuple,res)))
写法二:同样也是转化为3数之和,但这个想法和我自己的不太一样,代码也简洁得多

class Solution:
    def fourSum(self, nums):
        nums.sort()
        length=len(nums)
        ans=[]
        for i in range(length):
            for j in range(i+1,length):
                l,r,t=j+1,length-1,target-nums[i]-nums[j]
                while l<r:
                    while l<r and t>nums[l]+nums[r]: l+=1   #此处nums[l]和nums[r]是在变化的
                    while l<r and t<nums[l]+nums[r]: r-=1
                    if l<r and nums[l]+nums[r]==t:
                        ans.append([nums[i],nums[j],nums[l],nums[r]])
                    l+=1
        return list(set(map(tuple,ans)))
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值