【Leetcode】之3sum问题

一.问题描述

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可以实现排序。/article/7913549.html

排完序之后就是寻找的问题，题目要求保证找到的都是不重复的组合，在这个地方需要好好想想，自己在这个点上折腾了一番。最后测试通过的程序如下：

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;

}
};