1. 程式人生 > 其它 >leetcode 18. 四數之和

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
}