leetcode 18. 四数之和

一、题目

给你一个由 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]]

二、解法

本题思路与三数之和一样，用双指针法，将暴力求解的\(O(n^4)\)降为\(O(n^3)\)
但本题不是求0，而是求target，target可能为负数，所以不能剪枝。

func fourSum(nums []int, target int) [][]int {
    len:=len(nums)
    if len<4{
        return nil
    }
    sort.Ints(nums)
    ans:=make([][]int,0)
    for i:=0;i<len;i++{
        if i>0&&nums[i]==nums[i-1]{
            continue
        }
        for j:=i+1;j<len;j++{
            if j>i+1&&nums[j]==nums[j-1]{
                continue
            }
            k,l:=j+1,len-1
            for k<l{
                sum:=nums[i]+nums[j]+nums[k]+nums[l]
                if sum==target{
                    ans=append(ans,[]int{nums[i],nums[j],nums[k],nums[l]})
                    k++
                    l--
                    for k<l&&nums[k]==nums[k-1]{
                        k++
                    }
                    for k<l&&nums[l]==nums[l+1]{
                        l--
                    }
                }else if sum>target{
                    l--
                }else{
                    k++
                }
            }
        }
    }
    return ans
}