双指针法解决三数之和问题

15. 三数之和

原理

利用一个循环控制第一个数，双指针控制后两个数；
将数组进行升序排列，比0大就调小一点，比0小就调大一点；
注意元素去重，防止最后结果出现一样的；
注意后两个数去重的位置，一定是在得到一次值以后才需要去重

JS实现

/**
 * @param {number[]} nums
 * @return {number[][]}
 */

// 双指针法
var threeSum = function (nums) {
  let results = []
  // 从小到大排序
  nums.sort((a, b) => a - b)
  for (let i = 0; i < nums.length; i++) {
    // 数组中最小的元素大于零，那么数组中不可能出现三个数之和为零
    if (nums[i] > 0) return results

    let left = i + 1
    let right = nums.length - 1
    // 第一个元素的去重，注意去重时是当前元素和之前元素相比，即和nums[i-1]相比
    if (i > 0 && nums[i] === nums[i - 1]) {
      continue
    }
    while (left < right) {
      if (nums[left] + nums[right] + nums[i] > 0) {
        // 当前三数之和大于零，需要把和减小一点
        right--
      } else if (nums[left] + nums[right] + nums[i] < 0) {
        // 当前三数之和小于零，需要把和增大一点
        left++
      } else {
        // 当前三数之和等于零
        results.push([nums[left], nums[right], nums[i]])
        // 注意去重的位置，一定是在得到一次值以后才需要去重
        // 第二个元素的去重，用过的数字就不要再用了
        while (left < right && nums[left] === nums[left + 1]) {
          left++
        }
        // 第三个元素的去重
        while (left < right && nums[right] === nums[right - 1]) {
          right--
        }
        left++
        right--
      }
    }
  }
  return results
}