给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。
样例
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是:
(-1, 0, 1)
(-1, -1, 2)
注意
在三元组(a, b, c),要求a <= b <= c。
结果不能包含重复的三元组
方法不太好,有了好方法会马上改进,请见谅!
class Solution {
public:
map<vector<int>,bool> check;//防止相同的序列放入结果中
void backtracing(vector<vector<int> > &ret,vector<int> &nums,vector<int> &vec,int cur,int n,int left,int count){
if(3==count){ //count为计数器,当累加数等于3时(只允许有三个候选值)
if(0==left&&false==check[vec]){ //left==0表示累加和已经等于0了,还需要判断该序列是否已经加入结果中
check[vec]=true; //加入就做了记号
ret.push_back(vec);
}
return ;
}
for(int i=cur;i<n;++i){
if(left>=nums[i]){
vec.push_back(nums[i]);
backtracing(ret,nums,vec,i+1,n,left-nums[i],count+1);
vec.pop_back();
}else //因为是递增序列,越往后数字越大,没必要往后检验了
break;
}
}
vector<vector<int> > threeSum(vector<int> &nums) {
vector<vector<int> > ret;
vector<int> vec;
if(nums.empty())
return ret;
sort(nums.begin(),nums.end());//先排序,减少很多工作
backtracing(ret,nums,vec,0,nums.size(),0,0);//第1个0表示nums的下标从0开始,第2个0表示S=0(三数和),第3个0(计数器从0开始)
return ret;
}
};