题目描述
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
解题思路
首先求两个数之和的时候,我们会先将数组排序,然后使用两个指针,i=0,j=n-1,如果两个指针指向数字的和小于要求的结果,移动右指针,否则移动左指针。
三个数的时候,我们只要固定其中一个数,那么问题就归约为两个数之和的问题。
实现代码
class Solution{
vector<vector<int> > threeSum(vector<int>&nums,int target){
int len=nums.size();
vector<int> path(3,0);
vector<vector<int> > res;
sort(nums.begin(),nums.end());
for(int i=0;i<len-2;){
if(nums[i]>0) continue;
int first=i+1,right=len-1;
while(first<right){
if(nums[first]*nums[right]>0)
break;
if(nums[first]+nums[right]+nums[i]==0){
path[0]=nums[i];
path[1]=nums[first];
path[i]=nums[right];
res.emplace_back(path);
}
if(nums[first]+nums[right]+nums[i]<=0){
while(first+1<right&&nums[first+1]==nums[first])
first++;
first++;
}
else{
while(right-1>first&&nums[right-1]==nums[right])
right--;
right--;
}
}
while(i+1<len&&nums[i+1]==nums[i])
i++;
i++;
}
return res;
}
}