这道题花费了我很长时间,一直不能通过。最后参照了discuss的热门答案,终于解决了。
我的思考过程:
暴力法:先是学combination sum,用递归解决,后来发现复杂了,其实就是三重循环O(n*n*n)
用二分查找优化:排序,前两个二重循环,第三个数二分查找,复杂度是O(n*n*logn)
一直通不过,然后百度之,看到了O(n*n)的算法。
之前先找了:编程之美 2.12 快速寻找满足条件的两个数 ,看懂了,就好解决这个问题。
Ps:注意去重的简洁方法,我之前还专门为去重写了个函数,也超时了。不过现在的方法的确简洁
//error1:未注意数组中的数字只能用1次
//error2:数组中可能出现重复的数字,所以结果中出现了一样的结果
//error3:保存中间结果的数组定义位置有问题导致了其大小超出了3
//error4:未考虑周全数组出现重复数字的影响
vector<vector<int> > threeSum(vector<int> &num) {
sort(num.begin(), num.end());
vector<vector<int> > result;
for (int i = 0; i < num.size(); i++)
{
int target = 0 - num[i];
int k = num.size() - 1;
int j = i + 1;
while ( j<k )
{
int low = num[j];
int high = num[k];
int sum = low + high;
if (sum < target)
{
j++;
continue;
}
else if (sum>target)
{
k--;
}
else
{
vector<int> midResult;
midResult.push_back(num[i]);
midResult.push_back(num[j]);
midResult.push_back(num[k]);
result.push_back(midResult);
while (j<k && (num[j] == midResult[1]) )//j去重
{
j++;
}
while (j<k && (num[k] == midResult[2]) )// k去重
{
k--;
}
}
}
//去i重
while ( (i+1<num.size()) && (num[i+1] == num[i]))
{
i++;
}
}
return result;
}