给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意: 答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
思路
拿数组nums = [-1, 0, 1, 2, -1, -4]来举例,首先将数组排序,然后有一层for循环,i从下标0的地方开始,同时定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。
依然还是在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i],b = nums[left],c = nums[right]。
接下来如何移动left 和right呢,如果nums[i] + nums[left] + nums[right] > 0 就说明此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。
如果 nums[i] + nums[left] + nums[right] < 0 说明此时三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。
如果nums[i] + nums[left] + nums[right] == 0就证明找到了满足条件的三个数,直接输出即可。但是此时还需考虑b c两个值的去重,只有abc三个数的去重操作都执行了结果才不会有重复的元组。
代码:
//三数之和 a+b+c=0,元组不重复;
#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
int num[] = {-1,0,1,2,-1,-4};
sort(num,num+6);
for(int i=0;i<6;i++)
{
if(num[i]>0) break; //剪枝
if(i>0 && num[i]==num[i-1]) continue; //a的去重操作
int left = i+1;
int right = 5;
while(right>left)
{
if(num[i]+num[left]+num[right]<0) left++;
else if(num[i]+num[left]+num[right]>0) right--;
else
{
cout<<num[i]<<" "<<num[left]<<" "<<num[right]<<endl;
while(right>left && num[left]==num[left+1]) left++; //去重b
while(right>left && num[right]==num[right-1]) right--; //去重c
left++;
right--;
}
}
}
return 0;
}