题目
给你一个包含 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]
输出:[]
分析
三数之和是在二数之和的基础上进行的变动,其实就是再增加了一层循环,最好还是不要用哈希表,因为还要去重就会很麻烦。先进行排序,对每一个i值寻找另外两个数字和他相加等于目标值。
代码
var threeSum = function(nums) {
let n = nums.length
if (n<3) return []
let res = []
nums.sort((a,b)=>a-b)
for(let i=0; i<n; i++){
if(i>0 && nums[i]===nums[i-1]) continue
let left = i+1, right = n-1
while(left<right){
let sum = nums[i]+nums[left]+nums[right]
if(sum===0){
res.push([nums[i],nums[left],nums[right]])
left++
while(nums[left] === nums[left-1]) left++
}else if(sum>0){
right--
}else{
left++
}
}
}
return res
};