Leetcode 18:四数之和(最详细解决方案!!!)

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

注意:

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

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

解题思路

这个问题其实和Leetcode 15:三数之和(最详细解决方案!!!)类似,所以我们可以直接参考这篇文章中的做法。

我们首先想到的做法就是暴力解法

class Solution:
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        result = []
        for i, a in enumerate(nums):
            for j, b in enumerate(nums[i + 1:]):
                for k, c in enumerate(nums[j + i + 2:]):
                    for _, d in enumerate(nums[i + j + k + 3:]):
                        if a + b + c + d == 0:
                            result.append([a, b, c, d])
        return result

暴力解法是如此的简洁( ̄▽ ̄)",但是正如大家所知,这种做法的时间复杂度时O(n^4)级别的。我们能不能优化为O(n^3)级别呢?

我们可以将d中的元素都放入查找表,这样我们只需要遍历a\b\c三个数组,之后在查找表中查找0 - a - b - c就可以将问题解决,我这里就不再赘述地代码。

我们还不满足,有没有更快的解法?有!!!

我们可以参考Leetcode 15:三数之和(最详细解决方案!!!)文章中的对撞指针的思路

class Solution:
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        result = list()
        nums_len = len(nums)
        nums.sort()
        for i in range(nums_len - 3):
            
  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值