LeetCode 18 4Sum (Tag:Array Difficulty:Medium)

前言

关于 LeetCode 数组类型题目的相关解法，可见LeetCode 数组类型题目做前必看，分类别解法总结了题目，可以用来单项提高。觉得有帮助的话，记得多多点赞关注哦，感谢！

题目描述

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

链接：leetcode-cn.com/problems/4s…

题解

1. 暴力解法，四层循环，并去重，时间复杂度为 O(n4)
2. 双层循环后加双指针减少时间复杂度，同时事先排序，用于去重，见代码，时间复杂度 O(n³)

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[][]}
 */
var fourSum = function(nums, target) {
    if (nums.length < 4) return []
    nums.sort((a,b) => a - b)
    const n = nums.length
    let res = []
    for (let i = 0 ; i < n - 3; ++i) {
      if (i > 0 && nums[i] === nums[i-1]) continue  // 去重
      for (let j = i + 1; j < n - 2; ++j) {
        if (j > i + 1 && nums[j] === nums[j-1]) continue // 去重
        let l = j + 1
        let r = n - 1
        const leave = target - nums[i] - nums[j]
        while (l < r) {
          if (nums[l] + nums[r] === leave) {
            res.push([nums[i], nums[j], nums[l], nums[r]])
            while (l < r && nums[l] === nums[l+1]) ++l // 去重
            while (l < r && nums[r] === nums[r-1]) --r // 去重
            ++l
            --r
          } else if (nums[l] + nums[r] > leave) {
            --r
          } else {
            ++l
          }
        }
      }
    }
  return res
};