题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题思路
方法一 暴力求解
由题干可知,我们需要寻找的是三个数,那就使用三重循环嵌套最为直观
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
let arr = []
for (var i = 0; i < nums.length - 2; i++) {
for (var j = i + 1; j < nums.length - 1; j++) {
for (var k = j + 1; k < nums.length; k++) {
if (nums[i] + nums[j] + nums[k] === 0) {
var tmp = [nums[i] , nums[j] , nums[k]];
arr.push(tmp)
}
}
}
}
return arr
};
我们虽然能够通过这种遍历方式找到答案,但是时间复杂度很高,并且不能直接的实现去重复的操作,还需在做运算。因此,除非特殊情况,我们不太应该使用这种方式。
方法二 双指针遍历
我们在此处也可以使用双指针遍历的方法找到结果。
- 首先将数组元素进行从小到大的排序。我们首先固定最左侧的一个元素
cur
作为基准,然后使用左右指针指向cur
后面的两端,分别为left
和right
。计算nums[cur],nums[left],nums[right]
的和是否为0,满足则添加到结果集。 - 如果
nums[cur]
大于零,由于数组是从小到大排序,那么三数之和一定大于0,此时结束循环。 - 如果
nums[cur] == nums[cur-1]
则说明基准数重复,会导致结果重复,此时跳过循环。 - 当和为零时,如果
nums[left] == nums[left+1]
则结果重复,跳过循环。 - 当和为零时,如果
nums[right] == nums[right + 1]
则结果重复,跳过循环。
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
let res = [];
const len = nums.length;
if(nums == null || len < 3) return res;
nums.sort((a,b)=>a-b);
for (let cur = 0; cur < len ; cur++) {
if(nums[cur] > 0) break; // 如果当前数字大于0,则三数之和一定大于0,所以结束循环
if(cur > 0 && nums[cur] == nums[cur-1]) continue; // 去重
var left = cur+1;
var right = len-1;
while(left < right){
var sum = nums[cur] + nums[left] + nums[right];
if(sum == 0){
res.push([nums[cur],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 if (sum > 0) right--;
}
}
return res;
};