【Daily Interview】- 24 三数之和

381 阅读1分钟

题目

img01

题目来源:三数之和 - 力扣

分析

这道题是两数之和的进阶,感兴趣的读者可以先去看一看。

在前面的文章中已经提到过,对于数组问题,通常会使用双指针来解决问题,这里也不例外。

首先题目中要求不能出现重复的三元组,对于这个要求,我们可以借助额外的空间来进行判断,而这里使用方法是:先排序,然后在找答案的过程中跳过重复的元素。

那么这里如何使用双指针呢?

需求需要找 3 个数,那么这里我们可以固定一个数,然后用指针分别从首尾找另外两个数,如下图所示:

固定 index,通过 left 和 right 遍历数组余下部分,随后 index 自增,left 和 right 恢复初始位置继续遍历就可以完整遍历整个数组所有三数相加之和的结果了。

img02

这里需要注意的有:

  • left 初始值为 index + 1
  • left 不能大于 right

那么整个方法的宏观框架如下:

for (let i = 0; i < len; i++) {
    let left = i + 1;
    let right = len - 1;
    if (nums[i] === nums[i - 1]) continue;
    while (left < right) {
      /* do something */
    }
  }

在遍历两个指针的过程中,有以下几种情况:

  • nums[i] + nums[left] + nums[right] == 0:说明找到了,直接将结果添加进结果集,并且跳过之后的重复元素
  • nums[i] + nums[left] + nums[right] > 0 :因为数组已经经过了排序,所以缩小 right 即可
  • nums[i] + nums[left] + nums[right] < 0 :同上,增大 left 即可

那么在 while 循环中,代码大致如下:


if (nums[left] + nums[right] + nums[i] === 0) {
  result.push([nums[left], nums[right], nums[i]]);
  while (nums[left] === nums[left + 1]) left++;
  left++;
  while (nums[right] === nums[right - 1]) right--;
  right--;
} else if (nums[left] + nums[right] + nums[i] > 0) {
  right--;
} else {
  left++;
}

结果如下:

img03