给你一个包含 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]
输出:[]
题解
- 方法一:(执行效率非常低。。)
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
const len = nums.length;
if (len < 3) return []; // 如果长度小于3,不存在三元组
nums.sort((a, b) => a - b); // 升序排序
if(nums[0] > 0) return []; // 如果最小数大于0,不存在和等于0
let result = [];
for(let i = 0; i < len - 2; i++) {
if(nums[i] > 0) break; // 如果三元组中最小数大于零,结束循环
for(let j = i + 1; j < len - 1; j++) {
let n = nums[i] + nums[j];
if(n > 0) break;
else n = -n;
if(nums.indexOf(n, j + 1) !== -1) { //查看其和是否存在数组中
result[result.length] = [nums[i], nums[j], n];
}
while(nums[j] == nums[j + 1]) j++; // 去重
}
while(nums[i] == nums[i + 1]) i++; // 去重
}
return result;
};
- 方法二:双指针
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
nums.sort((a, b) => a - b);
let result = [];
let len = nums.length;
for(let i = 0; i < len - 2; i++) {
// 如果三元组中最小数大于零,结束循环
if(nums[i] > 0) return result;
if(i > 0 && nums[i-1] == nums[i]) continue;
let n = nums[i];
let left = i + 1; // 左指针
let right = len - 1; // 右指针
while(left < right) {
let l = nums[left];
let r = nums[right]
let sum = n + l + r;
if(sum == 0) {
result[result.length] = [n, l, r];
while (left < right && l == nums[left]) left++; // 直到指向不一样的数
while (left < right && r == nums[right]) right--; // 直到指向不一样的数
}
else if(sum > 0) {
right--;
}
else {
left++;
}
}
}
return result;
};