一.问题描述
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.
二.我的解题思路
之前做过2sum的问题,所以自然是想能否借用2sum的思路。本题需要求解的是三个数和为定值,现在固定其中一个数,那么就可以转化为求解2sum的问题了。之前是用hash表来做2sum的问题,所以本题一开始也这样去做,但是写完程序结果超时了。
仔细分析题意,题目中特别指出要求解的组合是不重复的,而且每个组合内部是不减的。用hash表可以做出来,但是需要考虑很多边界条件才能满足不重复这点。所以最好换个思路,先排序,再用指针夹逼。
c++ vector有现成的sort可以实现排序。http://blog.youkuaiyun.com/ajioy/article/details/6976945
排完序之后就是寻找的问题,题目要求保证找到的都是不重复的组合,在这个地方需要好好想想,自己在这个点上折腾了一番。最后测试通过的程序如下:
class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
vector<vector<int> > ret;
ret.clear();
sort(num.begin(),num.end());
for(int i=0; i!=num.size();i++){
if(i > 0 && num[i]==num[i-1])
continue;
int j,k;
j=i+1;
k=num.size()-1;
while(j<k){
if(j>i+1&&num[j]==num[j-1]){
j++;
continue;
}
if(k<num.size()-1&& num[k]==num[k+1]){
k--;
continue;
}
int sum = num[i] + num[j] + num[k];
if(sum>0){
k--;
}else if(sum<0){
j++;
}else{
vector<int> tmp;
tmp.push_back(num[i]);
tmp.push_back(num[j]);
tmp.push_back(num[k]);
ret.push_back(tmp);
j++;
}
}
}
return ret;
}
};