python3刷LeetCode-18.四数之和

题目描述
18. 四数之和
给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:答案中不可以包含重复的四元组。

示例 1:

输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:

输入:nums = [], target = 0
输出:[]

题目链接

代码:

class Solution:
    def fourSum(self, nums: list, target: int) -> list:
        n = len(nums)
        nums.sort()
        res = []
        flag = 0
        for k in range(n-3):
            if k > 0 and nums[k] == nums[k-1]:
                continue
            for l in range(k+1, n-2):
                if l > 1 and nums[l] == nums[l-1] and l-1 != k:
                    continue
                i = l+1
                j = n-1
                while i < j:
                    sum = nums[i]+nums[j]+nums[l]+nums[k]
                    if sum < target:
                        i += 1
                    elif sum > target:
                        j -= 1
                    else:
                        res.append([nums[k], nums[l], nums[i], nums[j]])
                        i += 1
                        j -= 1
                        while i < j and nums[i] == nums[i-1]:
                            i += 1
                        while i < j and nums[j] == nums[j+1]:
                            j -= 1
        return res


if __name__ == '__main__':
    s = Solution()
    print(s.fourSum([-4,-3,-2,-1,0,0,1,2,3,4],0))

解题思路
排序+双指针
基本思路:这题解法和三数之和一样,比三数之和多了一重循环,第一重循环k从0到n-4,第二重循环 l 从k+1到n-3(下标从0开始,所以是到n-3)。接下来就是和三数之和一样的解法了,双指针,i 从 l+1 开始,j从n-1开始,计算和sum,小于target则i+1,大于target则j-1,等于target把该组数加入到结果中,执行i+1,j-1,选取下一组数。
去重:四数之和的去重第一重循环和里面的双指针去重方式是一样的,直接与上一个状态相比较,相同就跳过。第二重循环去重需要判断一下 l 是否大于k+1再进行比较,这样才能确保进行比较的是选取的四元组中的第二个数,而不是与k对应的值(即该轮选取的四元组的第一个数)比较。
时间复杂度:双重循环加双指针,时间复杂度是O(N^3),排序的时间复杂度是O(nlogn),所以总的时间复杂度是O(N ^3+nlogn),小的数量级忽略,即O(N ^3)。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值