给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
先固定第一个数,再用双指针确定后面两个数。
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
sort(nums.begin(),nums.end());
/*
size_t为无符号数,如果nums.size()<2,则nums.size()-2会溢出,
因此nums.size()必须强制转换为int类型
*/
for(int i=0;i<(int)nums.size()-2;++i){
//剪枝操作:如果第一个数就大于0那么三数之和肯定不为0
if(nums[i]>0) break;
//保证不包含重复三元组
if(i>0&&nums[i]==nums[i-1]) continue;
int l=i+1,r=nums.size()-1;
//双指针
while(l<r){
if(nums[l]+nums[r]==(-1)*nums[i]){
res.push_back({nums[i],nums[l],nums[r]});
//保证不包含重复二元组
while(l<r&&nums[l+1]==nums[l]) l++;
while(l<r&&nums[r]==nums[r-1]) r--;
l++;
r--;
}
else if(nums[l]+nums[r]<(-1)*nums[i]) l++;
else r--;
}
}
return res;
}