题5:三数之和

127 阅读1分钟

题5:三数之和

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

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

示例 1:

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]

示例 2:

输入:nums = []
输出:[]

示例 3:

输入:nums = [0]
输出:[]

提示:

0 <= nums.length <= 3000
-105 <= nums[i] <= 105
分析:
  1. 找到的正确结果都返回
  2. 结果里不能有相同的数组
解题思路:
  1. 给数组排序
  2. 遍历数组,从0遍历到length-2(防止越界,保证同时能放下三个指针 i, start, end)
  3. 如果当前数字等于前一个数字,则跳过这个数(去重,防止有相同的结果出现)
  4. 如果数字不同,则设置start = i + 1,end = length - 1,查看i,start,end 三个数的和比0大,还是比0小,如果比0小,start++,如果比0大,end--,如果等于0,把这三个数加入到结果里。
  5. 返回结果
图解:

Snipaste_2021-08-11_14-26-20.jpg

代码:

<script>
    var threeSum = function (nums) {
      const result = [];
      nums.sort(function (a, b, c) {
        return a - b;
      })
​
      for (let i = 0; i < nums.length - 2; i++) {
        if (i === 0 || nums[i] !== nums[i - 1]) {
          let start = i + 1, end = nums.length - 1;
          while (start < end) {
            if (nums[i] + nums[start] + nums[end] === 0) {
              result.push([nums[i], nums[start], nums[end]]);
              start++;
              end--;
              while (start < end && nums[start] === nums[start - 1]) {
                start++;
              }
              while (start < end && nums[end] === nums[end + 1]) {
                end--;
              }
            } else if (nums[i] + nums[start] + nums[end] < 0) {
              start++;
            } else {
              end--;
            }
          }
​
        }
      }
    };
  </script>