leetcode不定期刷题---18. 四数之和

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

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 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]
]

题解：

会了之前三数之和，这条题目就会变得简单了，只需要在三数之和外面再套一层循环。

思路：

数组先进行排序。从小到大排序。循环最外层，相当于固定第一个数字，在进行一边循环固定第二个数字，然后在剩下的数中找到和为 target-nums[i]-nums[j]的两个数字，最后组成相应的数组。

同时去重的操作要注意。第一层循环的时候，当nums[i] > Math.abs(target) || nums[i] === nums[i-1]直接进入下一层循环等等。不过我这个方法虽然通过了，但是效率很低。

var fourSum = function(nums, target){
        nums = nums.sort((a,b)=>{return a-b});
        let arr = [];
        if(!nums || nums.length<4)return arr;
        for(let i = 0; i < nums.length-3;i++){
            if(nums[i] > Math.abs(target) || nums[i] === nums[i-1])continue;
            for(let j = i + 1; j < nums.length-2;j++){
                if(nums[j] === nums[j-1] && (j-1) !== i)continue;
                let sum = target - nums[i] - nums[j];
                let left = j+1,right = nums.length-1;
                while(left < right){
                    if(nums[left]+nums[right]>sum){
                        do{--right}while(nums[right]===nums[right+1])
                    }else if(nums[left]+nums[right] < sum){
                        do{++left}while(nums[left]===nums[left-1]);
                    }else{
                        let newArr = [nums[i],nums[j],nums[left],nums[right]];
                        arr.push(newArr);
                        do{--right}while(nums[right]===nums[right+1]);
                        do{++left}while(nums[left]===nums[left-1])
                    }
                }
            }
        }
        return arr;
    };

执行结果：
执行用时 :124 ms, 在所有 javascript 提交中击败了85.26% 的用户
内存消耗 :36.5 MB, 在所有 javascript 提交中击败了81.40%的用户

然后进行优化，除去一些情况之后

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[][]}
 */
var fourSum = function(nums, target) {
      nums = nums.sort((a,b)=>{return a-b});
        let arr = [],n = nums.length;
        if(!nums || nums.length<4)return arr;
        for(let i = 0; i < nums.length-3;i++){
            if(nums[i] > Math.abs(target) || nums[i] === nums[i-1])continue;

            let target1 = target - nums[i];
            if (nums[i + 1] + nums[i + 2] + nums[i + 3] > target1) break;
            if (nums[n - 3] + nums[n - 2] + nums[n - 1] < target1) continue;

            for(let j = i + 1; j < nums.length-2;j++){
                if(nums[j] === nums[j-1] && j>i+1)continue;

                let sum = target1 - nums[j];
                if (nums[j + 1] + nums[j + 2] > sum) break;
                if (nums[n - 2] + nums[n - 1 < sum]) continue;

                let left = j+1,right = nums.length-1;
                while(left < right){
                    if(nums[left]+nums[right]>sum){
                        right--;
                    }else if(nums[left]+nums[right] < sum){
                        left++;
                    }else{
                        let newArr = [nums[i],nums[j],nums[left],nums[right]];
                        arr.push(newArr);
                        do{--right}while(nums[right]===nums[right+1]);
                        do{++left}while(nums[left]===nums[left-1])
                    }
                }
            }
        }
        return arr;
    }

执行结果
执行用时 :96 ms, 在所有 javascript 提交中击败了99.20% 的用户
内存消耗 :36.2 MB, 在所有 javascript 提交中击败了94.57%的用户

就是感觉代码量变大了很多，但是性能方面还行。