给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
双指针+排序
vector<vector<int>> threeSum(vector<int>& nums)
{
int len = nums.size();
vector<vector<int>> resVec;
if(len<3)
return resVec;
sort(nums.begin(),nums.end()); //排序
for(int i = 0;i<len;++i)
{
if(nums[i]>0) //当前值大于0,之后组合的值之和不可能等于0
break;
if(i>0 && nums[i] == nums[i-1]) //去重
continue;
int nL = i+1; //左“指针”(记录左下标)
int nR = len -1; //右“指针”
while(nL<nR)
{
int sum = nums[i] + nums[nL] + nums[nR];
if(sum == 0)
{
resVec.push_back({nums[i],nums[nL],nums[nR]});
while(nL<nR && nums[nL] == nums[nL+1]) //去重
++nL;
while(nL<nR && nums[nR] == nums[nR-1]) //去重
--nR;
++nL;
--nR;
}
else if(sum < 0)
++nL;
else if(sum > 0)
--nR;
}
}
return resVec;
}
int main()
{
vector<int> vec{-1,0,1,2,-1,-4};
vector<vector<int>> resVec = threeSum(vec);
for(auto &item : resVec)
{
for(auto &itemSub : item)
{
cout<<itemSub<<endl;
}
}
system("pause");
return 0;
}