15.三数之和

三数之和

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0

提示：

• 3 <= nums.length <= 3000
• $-10^5$ <= nums[i] <= $10^5$

解析

这题可以参考两数之和，是一个双指针的问题，选定一个元素，剩下的元素进行对向双指针求和。
因为最后求的是三元组的数值，所以可以先进行排序。

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function (nums) {
  const result = [];
  nums = nums.sort((i, j) => i - j);
  for (let i = 0; i < nums.length - 2; i++) {
    const a = nums[i];
    let start = i + 1;
    let end = nums.length - 1;
    // 去重
    if (i > 0 && a === nums[i - 1]) {
      continue;
    }
    while (start < end) {
      // 边界条件优化，如果最小的三个值都比0大，那可以直接 break
      if (a + nums[start] + nums[start + 1] > 0) {
        break;
      }
      // 边界条件优化，如果最大的三个值都比0小，那可以直接 break
      if (a + nums[end] + nums[end - 1] < 0) {
        break;
      }
      const sum = a + nums[start] + nums[end];
      // 去重
      if (start > i + 1 && nums[start] === nums[start - 1]) {
        start++;
        continue;
      }
      if (sum === 0) {
        result.push([a, nums[start], nums[end]]);
        start++;
      } else if (sum > 0) {
        end--;
      } else {
        start++;
      }
    }
  }
  return result;
};