LeetCode18.四数之和 JavaScript

713 阅读1分钟

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

答案参考:

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[][]}
 */
var fourSum = function(nums, target) {
    nums=nums.sort(function(a,b){return a-b});//先排序
    var arr=[];
    for(i=0;i<nums.length-3;i++){//第一个
        if (i>0 && nums[i-1]==nums[i]) continue
        for(j=i+1;j<nums.length-2;j++){//第二个
             if (j>i+1 && nums[j-1]==nums[j]) continue
            var k=nums.length-1
            var c=j+1;
            while(c<nums.length-1&&c!=k){//第三个
                var sum=nums[i]+nums[j]+nums[c]+nums[k]
                    if (c>j+1 &&nums[c]== nums[c-1]){
                        c++;                       
                        continue;
                    }
                    if (k<nums.length-1 &&nums[k]==nums[k+1]){
                        k--;                       
                        continue;
                    }
                if(sum==target){        
                        arr.push([nums[i],nums[j],nums[c],nums[k]]);         
                        c++
                        k=nums.length-1
                }
                else if(sum<target){
                    c++
                }
                else{
                    k--
                }              
            }
        }
    }   
    return arr
};

欢迎关注