题目描述
给定一个长度为n的数组,找出3个数a,b,c,使之满足a+b+c=0这个等式。找出满足这个条件的所有不重复集合。原题以及例子如下,
题目分析
PS:答案的集合不要求是有序的。
如果是暴力破解的话,那么这道题的时间复杂度为O(n^3),穷举出所有的例子,但肯定是会超时的。我们可以做出这么一个转换,a+b=-c。做过Two Sum的朋友,肯定一下子就能想到如何以O(n)的时间复杂度来解决这个问题。那么这道题便迎刃而解了。
我在这里简单介绍下,如何以O(n)的时间复杂度解决a+b=-c这个问题。有两种方法,一是直接使用STL中的HashMap,这个我就不介绍了,相信大家看了讨论区的答案一下就知道了;二是使用两个指针的方法。在这里我们假定已知的数组已经是有序的(这里的时间复杂度为O(n^2),所以可以在开始之间进行一个快排)。用两个指针,分别指向数组的头和尾,如果tail+head<-c,那么便把头指针想后移动;如果tail+head>-c,那么就把尾指针向前移动。显然,当我遍历完该数组后,便可得到所有的答案。此题解答源码如下,
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int> > res;
bool findOne = false;
int l = nums.size();
for (int i = l - 1; i >= 2; --i) {
if (i != l - 1 && nums[i] == nums[i + 1]) {
continue;
}
int goal = -nums[i];
findOne = false;
int first = 0;
int last = i - 1;
int preFirst = 0;
while (first != last) {
if (nums[first] + nums[last] < goal) {
++first;
} else if (nums[first] + nums[last] > goal) {
--last;
} else {
if (findOne && (preFirst == nums[first])) {
++first;
continue;
}
vector<int> temp = {nums[first], nums[last], nums[i]};
res.push_back(temp);
preFirst = nums[first];
++first;
findOne = true;
}
}
}
return res;
}
};