leetcode hot100 15.三数之和

85 阅读1分钟

image.png

思路:

排序+双指针。难点主要在于如何去除重复解。

拿这个nums数组来举例,首先将数组排序,然后有一层for循环,i从下标0的地方开始,同时定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。

依然还是在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i] b = nums[left] c = nums[right]。

接下来如何移动left 和right呢, 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。

如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。

image.png

代码:

var threeSum = function (nums) {
  if (nums.length < 3) return [];
  nums.sort((a, b) => { return a - b })
  let res = [];
  for (let i = 0; i < nums.length; i++) {
    if (nums[i] > 0) break;//数组有序,当前值大于一,组成的和必大于0,直接返回结果
    if (nums[i] == nums[i - 1]) continue;//去重
    let left = i + 1, right = nums.length - 1;
    while (left < right) {
      let sum = nums[left] + nums[right] + nums[i];
      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) left++;// 和小于0,左指针后移,让三数之和大一点
      else if (sum > 0) right--;// 和大于0,右指针前移,让三数之和小一点
    }
  }
  return res;
};