leetcode-18-四数之和

持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第15天，点击查看活动详情

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

提示：

1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

解题思路

本题和 leetcode-15-三数之和类似，只不过变成了四数之和且目标和值不再是固定的 0，而是传入的 target。
首先对数组进行排序，这样做的好处是可以极大缩小查找的范围，比如如果当前位置的值大于 target 且当前位置的值 >=0,那么它和它后面的值肯定无法组合成目标数组。
还有一个好处是如果当前元素和前一个元素相同，则跳过当前元素的处理，因为结果数组中不允许有同样的组合，如果当前元素和它前面的元素相同，则如果前面元素有符合条件的组合，它也会有，而且会相同，所以跳过。
利用以上两种策略，就可以极大程度的减少操作次数。

代码实现

var fourSum = function(nums, target) {
    const len = nums.length

    if(len<4){
        return []
    }

    let res = []
    

    nums.sort((a,b) => a-b)

    for(let i = 0;i<len-3;i++){
        if(nums[i]>target && nums[i]>=0){
            return res
        }

        if(i>0 && nums[i]===nums[i-1]){
            continue
        }

        for(let j = i+1;j<len-2;j++){
            if(nums[i]+nums[j] > target && nums[j]>=0){
                break
            }

            if(j>i+1 && nums[j]===nums[j-1]){
                continue
            }

            let head = j+1
            let end = len-1

            while(head<end){
                if(end<len-1 && nums[end]===nums[end+1]){
                     end--
                     continue
                }

                if(head>j+1 && nums[head]===nums[head-1]){
                    head++
                    continue
                }

                const sum = nums[i]+nums[j]+nums[head]+nums[end]

                if(sum===target){
                    res.push([nums[i],nums[j],nums[head],nums[end]])
                }
                if(sum-target>0){
                    end--
                }else{
                    head++
                }
            }
        }
    }

    return res
}

至此我们就完成了 leetcode-18-四数之和

如有任何问题或建议，欢迎留言讨论！