LeetCode 15. 三数之和

sulingliang

于 2021-06-30 16:57:09 发布

阅读量90

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_32862143/article/details/118365022

版权

LeetCode 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

第二种方法与第一种方法的区别在于：将三数之和转成两数之和

第一种方法：

const threeSum = (nums) => {
  if(nums.length < 2) {
      return [];
  }
  nums.sort((a, b) => a - b);
  const res = [];
  for(let i = 0; i < nums.length - 2; i++) {
    if(nums[i] > 0) {
      break;
    }
    if(i >= 1 && nums[i] === nums[i - 1]) {
      continue;
    }
    let left = i + 1;
    let right = nums.length - 1;
    while(left < right) {
      const sum = nums[i] + nums[left] + nums[right];
      if(sum === 0) {
        res.push([nums[i], nums[left], nums[right]])
        while(left < right && nums[left] === nums[left + 1]) {
          left++;
        }
        while(left < right && nums[right] === nums[right - 1]) {
          right--;
        }
        left++;
        right--;
      } else if(sum > 0) {
        right--;
      } else {
        left++;
      }
    }
  }
  return res;
};

第二种方法

const threeSum = (nums) => {
  if(nums.length < 2) {
    return [];
  }
  nums.sort((a, b) => a - b);
  const res = [];
  for(let i = 0; i < nums.length - 2; i++) {
    if(nums[i] > 0) {
      break;
    }
    if(i >= 1 && nums[i] === nums[i - 1]) {
      continue;
    }
    let target = 0 - nums[i];
    let left = i + 1;
    let right = nums.length - 1;
    while(left < right) {
      if(nums[left] + nums[right] === target) {
        res.push([nums[i], nums[left], nums[right]])
        while(left < right && nums[left] === nums[left + 1]) {
          left++;
        }
        while(left < right && nums[right] === nums[right - 1]) {
          right--;
        }
        left++;
        right--;
      } else if(nums[left] + nums[right] < target) {
        left++;
      } else {
        right--;
      }
    }
  }
  return res;
};