剑指 Offer II 007. 数组中和为 0 的三个数

120 阅读1分钟

剑指 Offer II 007. 数组中和为 0 的三个数

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a ,b ,c ,使得 a + b + c = 0 ?请找出所有和为 0 且 不重复 的三元组。

没序数组 可以变成有序的, 存在重复项的,可以将他们特殊处理下

三个关键位置

if (nums[i] == nums[i - 1]) continue;

while (nums[left] == nums[++left]);

while (nums[right] == nums[--right]);

var threeSum = function (nums) {
  nums.sort((a, b) => a - b);
  let len = nums.length;
  let res = [];
  for (let i = 0; i < len; i++) {
    if (nums[i] == nums[i - 1]) continue;
    let left = i + 1,
      right = len - 1;
    let c = nums[i];
    while (left < right) {
      if (c + nums[left] + nums[right] == 0) {
        res.push([c, nums[left], nums[right]]);
        while (nums[left] == nums[++left]);
        while (nums[right] == nums[--right]);
      } else if (c + nums[left] + nums[right] < 0) {
        left++;
      } else {
        right--;
      }
    }
  }
  return res;
};