15.三数之和

题目

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c 使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

一. 末位二分法

首先将数组排序，然后使用嵌套遍历的方法获取前两个数，第三个数使用二分法进行查找。

js实现

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
  var answer = []

  nums.sort((a, b) => a-b)
  
  for (var i=0;i<nums.length;i++) {
    // 如果第一个数大于零，则后面的两个数必然也大于零
    if (nums[i] > 0) {
      break
    }
    // 排除重复
    if (i > 0 && nums[i] === nums[i-1]) {
      continue
    }
    for (var j=i+1;j<nums.length;j++) {
      if (nums[i] + nums[j] > 0) {
        break
      }
      if (j > i+1 && nums[j] === nums[j-1]) {
        continue
      }
      // 使用二分法查找第三个数
      var left = j + 1
      var right = nums.length - 1
      while (left <= right) {
        var mid = (left + right) / 2 | 0
        var sum = nums[i] + nums[j] + nums[mid]
        if (sum < 0) {
          left = mid + 1
        } else if (sum > 0) {
          right = mid - 1
        } else {
          answer.push([nums[i], nums[j], nums[mid]])
          break
        }
      }
    }
  }

  return answer
}

复杂度分析

时间复杂度：O(n²*log(n))

空间复杂度：O(n)

测试结果

✔ Accepted
  ✔ 313/313 cases passed (272 ms)
  ✔ Your runtime beats 60.83 % of javascript submissions
  ✔ Your memory usage beats 68.25 % of javascript submissions (46.6 MB)

二. 双指针法

先排序，然后我们遍历数组作为第一个，选定第一个后，我们使用两个指针分别指向左右两端，如果和小于零则移动左边的指针，大于零则移动右边的指针。

js实现

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
  var answer = []

  nums.sort((a, b) => a-b)
  
  for (var i=0;i<nums.length;i++) {
    // 如果第一个数大于零，则后面的两个数必然也大于零
    if (nums[i] > 0) {
      break
    }
    // 排除重复
    if (i > 0 && nums[i] === nums[i-1]) {
      continue
    }

    var left = i+1
    var right = nums.length - 1
    while(left < right) {
      var sum = nums[i] + nums[left] + nums[right]
      if (sum < 0) {
        left++
        while (left > i+1 && nums[left] === nums[left-1]) {
          left++
        }
      } else if (sum > 0) {
        right--
        while (nums[right] === nums[right+1]) {
          right--
        }
      } else {
        answer.push([nums[i], nums[left], nums[right]])
        left++
        while (left > i+1 && nums[left] === nums[left-1]) {
          left++
        }
        right--
        while (nums[right] === nums[right+1]) {
          right--
        }
      }
    }
  }

  return answer
}

复杂度分析

时间复杂度：O(n²)

空间复杂度：O(n)

测试结果

✔ Accepted
  ✔ 313/313 cases passed (184 ms)
  ✔ Your runtime beats 98.04 % of javascript submissions
  ✔ Your memory usage beats 76.14 % of javascript submissions (46.5 MB)