【LeetCode 18】四数之和

题目描述

给定一个包含 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]
]

解题思路

这个题实际上和三数之和类似，我们也使用双指针来解决。

在三数之和中，使用两个指针分别指向两个元素，左指针指向固定数后面的数，右指针指向最后一个数。在固定一个数，进行遍历。左指针不断向右移动，右指针不断向左移动，直至遍历完所有的数字。

在四数之和中，我们可以固定两个数字，然后再初始化两个指针，左指针指向固定数之后的数字，右指针指向最后一个数字。两层循环进行遍历，直至遍历完所有的结果。

需要注意的是，当使双指针的时候，往往需要对数组元素进行排序。

代码实现

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[][]}
 */
var fourSum = function(nums, target) {
    const res = []
    if(nums.length < 4){
        return []
    }
    nums.sort((a, b) => a - b)
    for(let i = 0; i < nums.length - 3; i++){
        if(i > 0 && nums[i] === nums[i - 1]){
            continue
        }
        if(nums[i] + nums[i +1] + nums[i + 2] + nums[i + 3] > target){
            break
        }

        for(let j = i + 1; j < nums.length -2; j++){
            // 若与已遍历过的数字相同，就跳过，避免结果中出现重复的数组
            if(j > i + 1 && nums[j] === nums[j - 1]){
                continue
            }
            let left = j + 1, right = nums.length - 1

            while(left < right){
                const sum = nums[i] + nums[j] +nums[left] + nums[right]
                if(sum === target){
                    res.push([nums[i], nums[j], nums[left], nums[right]])
                }
                if(sum <= target){
                    left ++
                    while(nums[left]  === nums[left - 1]) {
                        left ++
                    }
                }else{
                    right --
                    while(nums[right] === nums[right + 1]){
                        right --
                    }
                }
            }
        }
    }
    return res
    
};

提交结果