(15)3Sum
题目:在一个数组里面找到三个数字和等于0,输出所有的可能性,不要存在重复的组。
例子:提供的数组:s=[-1, 0, 1, 2, -1, -4],返回的结果应该是:[ [-1, 0, 1], [-1, -1, 2]]。
这道题本来想按原来两个数相加的那个来算,引用map,但是后来发现无法获取是否前面已经输出同样的结果,所以就很麻烦,或许有解决方法,但是暂时觉得可能会兜圈子,不如直接使用简单一点的排序方法。
首先直接sort一下vector,每次抽走最前面的一个数字,然后查找剩余的数组内是否有可以和为target=(0-抽走数字)的组合,通过left和right的查找方法,因为已经排好序了,如果和小于target,那么就应该left右移,这样能使得和变大,否则right左移,使得和变小,找到之后就记录一条,直到所有都记录完成。
避免重复的方法就显而易见了,由于都已经排序好了,那么每次找的时候只需要在一段子数组内有重复的跳过就好了,比如上面left右移和right左移的时候,只要是下一个跟当前数字相同,那么不运算直接移动一次,直到下一个数字和当前数字不同,才将当前数字放入运算中。
另外,如果暴力的三层循环搜索也就是O(n^3)的方法会导致超时TLE。
下面是代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int len = nums.size();
vector<vector<int>> s;
if(len == 0){
return s;
}
sort(nums.begin(), nums.end());
int target = 0;
int i, j, k;
vector<int> temp;
for(i=0;i<len;i++){
if(i!=0){
if(nums[i] == nums[i-1]){
continue;
}
}
target = 0;
target -= nums[i];
j = i + 1;
k = len - 1;
while(j<k){
if(nums[j] + nums[k] == target){
temp.push_back(nums[i]);
temp.push_back(nums[j]);
temp.push_back(nums[k]);
s.push_back(temp);
temp.clear();
while(j<k&&nums[j]==nums[j+1]){
j ++;
}
while(j<k&&nums[k]==nums[k-1]){
k --;
}
j++;
k--;
}
else if(nums[j] + nums[k] < target){
j ++;
}
else{
k --;
}
}
}
return s;
}
};