思路:
排序+双指针。难点主要在于如何去除重复解。
拿这个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相遇为止。
代码:
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;
};