leectcode15：三数之和

题目描述

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：

答案中不可以包含重复的三元组。
数组中的数字可以是小数或负数。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4]，

一个可能的解集为:

[  
  [-1, 0, 1],  
  [-1, -1, 2]  
]

解题思路

首先对数组进行排序，这样可以方便处理重复元素和减少搜索范围。
之后使用三个指针，i，left，right。其中 i 从数组的起始位置开始遍历，left 初始化为 i+1，right 初始化为数组末尾。

• 对于每个 i，固定它，然后移动 left 和 right 来查找满足 nums[i] + nums[left] + nums[right] == 0 的三元组。
• 如果和大于 0，则 right-- 以减小和；如果和小于 0，则 left++ 以增大和；如果和等于 0，则找到了一个解，并移动 left 和 right 来避免重复解，同时将它们添加到结果中。
• 在移动 left 和 right 时，需要跳过重复的元素以避免重复解。
  跳过重复元素：
• 在移动 i 时，如果 nums[i] 等于 nums[i-1]，则跳过该元素，以避免重复解。
• 在移动 left 时，如果 nums[left] 等于 nums[left-1]，则跳过该元素。
• 在移动 right 时，如果 nums[right] 等于 nums[right+1]，则跳过该元素。

解题示例 (JavaScript)

var threeSum = function(nums) {  
    const result = [];  
    nums.sort((a, b) => a - b); // 排序  
  
    for (let i = 0; i < nums.length - 2; i++) {  
        // 跳过重复元素  
        if (i > 0 && 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) {  
                result.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++;  
            } else {  
                right--;  
            }  
        }  
    }  
  
    return result;  
};