原题
Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
Note:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.
- 1
- 2
- 3
- 4
- 5
题目大意
给定一个n个元素的数组,是否存在a,b,c三个元素,使用得a+b+c=0,找出所有符合这个条件的三元组。
解题思路
可以在 2sum问题 的基础上来解决3sum问题,假设3sum问题的目标是target。每次从数组中选出一个数k,从剩下的数中求目标等于target-k的2sum问题。这里需要注意的是有个小的trick:当我们从数组中选出第i数时,我们只需要求数值中从第i+1个到最后一个范围内字数组的2sum问题。
我们以选第一个和第二个举例,假设数组为A[],总共有n个元素A1,A2….An。很显然,当选出A1时,我们在子数组[A2~An]中求目标位target-A1的2sum问题,我们要证明的是当选出A2时,我们只需要在子数组[A3~An]中计算目标位target-A2的2sum问题,而不是在子数组[A1,A3~An]中。
证明如下:假设在子数组[A1,A3~An]目标位target-A2的2sum问题中,存在A1 + m = target-A2(m为A3~An中的某个数),即A2 + m = target-A1,这刚好是“对于子数组[A3~An],目标位target-A1的2sum问题”的一个解。即我们相当于对满足3sum的三个数A1+A2+m = target重复计算了。因此为了避免重复计算,在子数组[A1,A3~An]中,可以把A1去掉,再来计算目标是target-A2的2sum问题。
对于本题要求的求最接近解,只需要保存当前解以及当前解和目标的距离,如果新的解更接近,则更新解。算法复杂度为O(n^2);
代码实现
#include <stdio.h>
#include <vector>
int Partition(int *v, const int low, const int high)
{
if (v == NULL || low > high) {
return -1;
}
int i = low;
int j = high;
int pivot = v[i];
while (i < j) {
while (i < j && pivot < v[j]) {
--j;
}
if (i < j) {
v[i++] = v[j];
}
while (i < j && pivot > v[i]) {
++i;
}
if (i < j) {
v[j--] = v[i];
}
}
v[i] = pivot;
return i;
}
void QuickSort(int *v, const int low, const int high)
{
if (v == NULL || low > high) {
return ;
}
if (low < high) {
int pivot = Partition(v, low, high);
QuickSort(v, low, pivot - 1);
QuickSort(v, pivot + 1, high);
}
}
bool Func(int *v, const int len, const int sum, std::vector<std::vector<int> > &vec)
{
bool ret = false;
if (v == NULL || len < 3) {
return ret;
}
QuickSort(v, 0, len - 1);
// 第一个加数
for (int i = 0; i < len - 2; ++i) {
// 第一个加数使用不能重复
if (i > 0 && v[i] == v[i - 1]) {
continue;
}
// 第二个加数
int start = i + 1;
// 第三个加数
int end = len - 1;
int n = sum - v[i];
while (start < end) {
if (v[start] + v[end] == n) {
std::vector<int> vecs;
vecs.push_back(v[i]);
vecs.push_back(v[start]);
vecs.push_back(v[end]);
vec.push_back(vecs);
// 保证再次使用第二个数不重复
do {
++start;
} while (start < end && v[start] == v[start - 1]);
// 保证再次使用第三个数不重复
do {
--end;
} while (start < end && v[end] == v[end + 1]);
} else if (v[start] + v[end] < n) {
// 保证再次使用第二个数不重复
do {
++start;
} while (start < end && v[start] == v[start - 1]);
} else {
// 保证再次使用第三个数不重复
do {
--end;
} while (start < end && v[end] == v[end + 1]);
}
} // end while
} // end i
return ret;
}
int main()
{
int v[] = {1,1,1,3,5,7,9,0,8,6,4,2};
const int len = sizeof(v) / sizeof(v[0]);
int sum = 10;
std::vector<std::vector<int> > vec;
Func(v, len, sum, vec);
QuickSort(v, 0, len - 1);
for (int i = 0; i < vec.size(); ++i) {
std::vector<int> vecs = vec[i];
for (int j = 0; j < vecs.size(); ++j) {
printf("%d ", vecs.at(j));
}
printf("\n");
}
return 0;
}
[root@agan tmall]# ./3sum
0 1 9
0 2 8
0 3 7
0 4 6
1 1 8
1 2 7
1 3 6
1 4 5
2 3 5