LeetCode数组第六题-四数之和

最新推荐文章于 2024-11-10 14:06:08 发布

月下梧桐静待花开

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量684

点赞数

分类专栏： LeetCode（天梯）记录文章标签： leetcode python

本文链接：https://blog.csdn.net/sinat_41860739/article/details/123915988

版权

LeetCode（天梯）记录专栏收录该内容

6 篇文章 0 订阅

订阅专栏

LeetCode数组第六题-四数之和

题目描述

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[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 = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum

这道题呢说实话经历了三数之和后非常之不难，同样是双指针的思路，只不过可以经过剪枝操作（减少不必要的计算）进一步优化，减少运行时间及复杂度
当然，我也没有一次成功，因为忽略了重复的数字，以至于有重复的解
代码如下：

class Solution(object):
    def fourSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[List[int]]
        """
        lens = len(nums)
        nums.sort()
        ans = []
        if lens >= 4:
            for i in range(lens-3):
                if i > 0 and nums[i] == nums[i - 1]:
                    continue
                if nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target:
                    break
                if nums[i]+nums[lens-1]+nums[lens-2]+nums[lens-3]<target:
                    continue
                for j in range(i+1,lens-2):
                    if j > i + 1 and nums[j] == nums[j - 1]:
                        continue
                    if nums[i]+nums[j]+nums[j+1]+nums[j+2]>target:
                        break
                    if nums[i]+nums[j]+nums[lens-1]+nums[lens-2]<target:
                        continue
                    low = j+1
                    high = lens-1
                    while low<high:
                        sums=nums[i]+nums[j]+nums[low]+nums[high]
                        if target == sums:
                            ans.append([nums[i],nums[j],nums[low],nums[high]])
                            while low<high and nums[low]==nums[low+1]:
                                low+=1
                            low+=1
                            while low<high and nums[high]==nums[high-1]:
                                high-=1
                            high-=1
                        elif target < sums:
                            high-=1
                        else:
                            low+=1
                    
        return ans