1、题目描述
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
2、解题思路
1、定义数组容器存放三元组
2、首先给定数组不为空且应至少有三个元素
3、将数组升序排列
4、遍历数组,找第一个加数a
4.1、若该数大于0则必然不满足条件,跳出循环
4.2、若该数重复出现则跳过,做去重处理
4.3、将加数a后的第一位数定义为第二个加数b,将最后一位数定义为第三个加数c
4.3.1、保持b在c前面的条件,定义sum为a、b、c之和
4.3.2、若和为0,则将三数作为一个三元组存入容器数组,并收缩a和b的距离,寻找另外的b、c与a搭配符合条件,同时要做去重判断
4.3.3、若和小于0,则右移b指针,找一个更大的b
4.3.4、若和大于0,则左移c指针,找一个更小的c
5、回到数组遍历,寻找下一个加数a,重复4的步骤
6、返回容器数组,即为符合条件的三元组数组
3、解题代码
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
let ans = [];
if(nums == null || nums.length < 3) return ans;
nums.sort((a, b) => a - b);
for (let i = 0; i < nums.length ; i++) {
if(nums[i] > 0) break;
if(i > 0 && nums[i] == nums[i-1]) continue;
let L = i+1;
let R = nums.length-1;
while(L < R){
const sum = nums[i] + nums[L] + nums[R];
if(sum == 0){
ans.push([nums[i],nums[L],nums[R]]);
while (L<R && nums[L] == nums[L+1]) L++;
while (L<R && nums[R] == nums[R-1]) R--;
L++;
R--;
}
else if (sum < 0) L++;
else if (sum > 0) R--;
}
}
return ans;
};