描述
给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。
在三元组(a, b, c),要求a <= b <= c。
结果不能包含重复的三元组。
样例
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是:
(-1, 0, 1)
给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。
在三元组(a, b, c),要求a <= b <= c。
结果不能包含重复的三元组。
样例
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是:
(-1, 0, 1)
(-1, -1, 2)
分析
因为题目说不能出现重复数组,所以我们需要排除这种情况,一旦发现有连续相等的数字,我们就跳过这个重复值。剩下的就比较简单了:
首先,我们需要对数组进行排序;然后,我们固定三个数之中的一个,让它从头到尾地遍历一次。对于剩下的两个值,让他们在第一个固定值到结尾最后一个值之间,通过三个数之后与目标判断其大小,好让剩下的两个值不断的从两端向中间逼近。最后,将与目标相等的数组压入vector中。
ps:下面代码也写了详细的注释,不懂的时候,最好在草稿上面试一试每次计算的结果,可以更方便地理解程序。
程序
class Solution {
public:
/**
* @param numbers: Give an array numbers of n integer
* @return: Find all unique triplets in the array which gives the sum of zero.
*/
vector<vector<int>> threeSum(vector<int> &numbers) {
// write your code here
vector<vector<int> > result;
sort(numbers.begin(), numbers.end());
for (int i = 0; i < numbers.size(); i++){
if (i>0 && numbers[i] == numbers[i - 1])// 数组{2,2,2,2,3,-2,0}
continue;
for (int j = i + 1, k = numbers.size() - 1; j < k;){
if (j > i + 1 && numbers[j] == numbers[j - 1]){// 数组{-4,-1,-1,-1,0,1,2}
j++;
continue;
}
if (k < numbers.size() - 1 && numbers[k] == numbers[k + 1]){// 数组{-4,0,1,2,-1,-1,-1}
k--;
continue;
}
int target = 0;
int sum = numbers[i] + numbers[j] + numbers[k];
if (sum < target)//小于target就增加
j++;
else if (sum > target)//大于target就减少
k--;
else//等于target就存入result中,存了之后记得k++,j--
{
vector<int> v(3, numbers[i]);
v[1] = numbers[j++];
v[2] = numbers[k--];
result.push_back(v);
}
}
}
return result;
}
};