题目:
给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
分析:
这道题刚看的时候想到的是先升序排序(方便保证结果唯一) 再用双指针法:首尾元素(nums[i] nums[j])相加若大于0 则让第三个元素nums[k]从i+1的位置开始遍历,直到找到三个数相加为0 的值或不满足边界条件,最后j--,当然要有边界条件。小于0则让k从j-1开始,最后i++。但是出了问题。。。等于0的时候我没搞通(因为不知最后是i动还是j动)。。
后来看了下别人的代码:为了确保不丢解,要再设层循环~在内层循环中可以按照刚刚的思路。。
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res ;
sort(nums.begin(),nums.end());
for(int i = 0 ; i < nums.size() ; i ++){
if(i > 0 && nums[i] == nums[i - 1]) continue;
int k = i + 1 , j = nums.size() - 1 , num ;
while(k < j){
num = nums[i] + nums[j];
if(num + nums[k] == 0){
res.push_back({nums[i] , nums[k] ,nums[j]}) ;
k ++ ; j -- ;
while(nums[k] == nums[k - 1] && k < j) k ++ ;
while(nums[j] == nums[j + 1] && j > k) j -- ;
}
else if(num + nums[k] > 0)
j -- ;
else
k ++ ;
}
}
return res ;
}
};