给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题思路:a+b+c=0可以转化为a+b=-c//我们可以先给这个数组排序,将第一个下标的值乘以-1保存起来,然后给两个指针,让第一个指针指向第二个数字,第二个指针指向最后,两个指针指向的值相加然后保存起来,如果与第一个保存的值相同的话,我们先开辟一个包含三个元素的数组,将这三个指向的值保存在这个数组里,然后判断重复的问题,如果第一个指针指向的值等于先前保存的指针指向的值的话,指针向后跑,如果第二个指针指向的值等于先前保存的指针指向的值的话,第二个指针向前跑,然后如果之前相加的和小于刚开始第一个下标的值的话,第一个指针向后跑,第二个指针向前跑,最后判断存放三元组中数组中第一个数有无重复,如果重复的话,下标加加.
//三元组x+y+z=0
vector<vector<int>> ThreeSum(vector<int> arr)
{
int len=arr.size();
vector<vector<int>> res;
for(int i=0;i<len;i++)
{
int sum=-1*arr[i];//x+y=-z//sum保存的-z
int left=i+1;//第一个指针:指向第二个元素
int right=len-1;//第二个指针:指向最后一个元素
while(left<right)
{
int sum1=arr[left]+arr[right];//x,y之和
if(sum==sum1)
{
vector<int> brr(3);//把三元组放在这个数组里面
brr[0]=arr[i];
brr[1]=arr[left];
brr[2]=arr[right];
res.push_back(brr);
while(left<right&&arr[left]==brr[1])//判断brr[1]重复
{
left++;
}
while(left<right&&arr[right]==brr[2])//判断brr[2]重复
{
right--;
}
}
else if(sum<sum1)//如果小于,++
{
left++;
}
else if(sum>sum1)//大于的话,--
{
right--;
}
}
while(i+1<len&&arr[i+1]==arr[i])//判断arr[0]是否重复
{
++i;
}
}
return res;
}
int main()
{
vector<int> arr;
for(int i=0;i<5;i++)
{
int tmp;
cin>>tmp;
arr.push_back(tmp);
}
vector<vector<int>> res=ThreeSum(arr);
int l=res.size();
for(int i=0;i<l;i++)
{
for(int j=0;j<3;j++)
{
cout<<res[i][j];
}
cout<<endl;
}
}
运行结果: