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.
For example, given array S = {-1 0 1 2 -1 -4}, A solution set is: (-1, 0, 1) (-1, -1, 2)首先,这题与2sum有着异曲同工之妙,只是难点在于排除重复的三元组。
以下是我的思路:
以{-1, -1, 0, 0, 0, 1, 2, 4}为例:首先,将所有数hash到一个hashmap中,unordered_map<int, int>,key为数,value为出现的次数。
假设hashtable的内容如下:
[-1 0 1 2 4] 数字
[2 3 1 1 1] 出现次数
两个指针i, j,j从0到最后,每次移动一个位置;i每一个循环中从0移到j;
每次循环中,i和j各用掉一个“出现次数”,那么就要判断0-num[i]-num[j]在hashtable中是否有出现次数,并且该数要大于num[j](由大到小找,防止重复)。
最后得出的结果就是不重复的所有sum三元组,这样,排序时间复杂度为O(nlgn),处理的时间复杂度为O(n^2)。
以下是源码(在leetcode上跑的结果是 超时,但我认为这个方法不错,就记录了下来)
class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
//排序
vector<vector<int>> res;
std::sort(num.begin(), num.end());
//hashtable统计
unordered_map<int, int> ht;
for(int k=0; k<num.size(); ++k) {
ht[num[k]]++;
}
//将num缩为不重复vector
vector<int>::iterator it = std::unique(num.begin(), num.end());
num.erase(it, num.end());
int p1, p2;
for(int j=0; j<num.size(); ++j) {
p2 = num[j];
for(int i=0; i<=j; ++i) {
unordered_map<int, int> tmp_ht(ht);
p1 = num[i];
tmp_ht[p1]--;
tmp_ht[p2]--;
int targetNumber = 0 - p1 - p2;
if(ht[targetNumber]!=0 && p2<=targetNumber) {
res.push_back({p1, p2, targetNumber});
}
}
}
return res;
}
};