一、题目描述
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。输出的顺序和三元组的顺序并不重要。
二、解题思路
1.求三个数之和,可以先确定一个数,然后就变成求两数之和的问题。
2.先将数组排序,开始循环数组,当前元素(nums[i])为固定值,在nums[i+1]~nums[nums.length-1]之间寻找剩下的两个值。思路与求两数之和差不多。
3.去重,因为要寻找到所有的组合,在指针移动的过程中,可能遇到相同的值,由于先前将数组排序,所以相同的值会相邻在一起,使用
while(left < right && nums[left] === nums[left+1]) left++
while(left < right && nums[right] === nums[right-1]) right--
将会得到相同的结果的值跳过。
三、解题代码
示例:
/**
* @param {number[]} nums
* @return {number[][]}
*/
var threeSum = function(nums) {
nums.sort((a,b)=>a-b)
let result = []
for(let i = 0;i<nums.length - 2;i++){
if(nums[i] > 0) break
// 去重,第一个数的去重
if(i > 0 && nums[i] === nums[i-1]) continue
let left = i + 1
let right = nums.length - 1
while(left < right){
let 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 if(sum > 0){
right--
}
}
}
return result
};
代码解析:注意一点细节:
1.固定数值的去重:if(i > 0 && nums[i] === nums[i-1]) continue ;只能与上一个数相比,因为下一个相同的数会在寻找两数之和的范围内,如果直接去掉会造成结果错误。
2.let sum = nums[i] + nums[left] + nums[right];在判断外确定,否则直接使用nums[i] + nums[left] + nums[right]表达式,在指针被修改的时候也会变化,导致结果不准确。