题目:
给你一个包含 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]]
思路:
从题目的要求很容易就可以知道,需要得到三个数就需要用到三个指针去指向这三个数,
1.首先给到我们的是一个无序的数组,所以我们为了后续的遍历查找,可以先将数组排序(升序或者降序都可以)
2.遍历数组,从0遍历到length-2 ,因为在遍历的过程中start的位置不能和end的位置相同,这样就会重复遍历,所以最终只能遍历到除去start和end两个指针下的长度 即是 length - 2
3.如果当前的数字等于前一个数字则跳过该数字,因为输出的结果不能有重复的两组数据
4.查看 i start 和 end 三个数的和并和 0 比较 若是等于 0 将这三个数以数组的形式添加到结果中,并且 start 右移 end 左移 , 如果小于0 则 start(end)右移(左移),如果大于0 则 start (end)
右移(左移)即是start++(end--) end--(start++)
5.返回结果
代码:
var threeSum = function(nums) {
const result = []
nums.sort(function(a , b) {
return a - b
})
for(let i = 0 ; i<nums.length - 2 ; i++) {
if(i === 0 || nums[i] !== nums[i-1]) {
let start = i + 1 , end = nums.length - 1
while(start < end) {
if(nums[start] + nums[i] + nums[end] === 0) {
result.push([nums[i] , nums[start] , nums[end]] )
start++
end--
while(start < end && nums[start] === nums[start-1]) {
start++
}
while(start < end && nums[end] === nums[end+1]) {
end--
}
} else if(nums[start] + nums[i] + nums[end] > 0) {
end--
} else {
start++
}
}
}
}
return result
};