之前做过一道两数之和的题目,可以用暴力法求解,也可以用双指针求解,更方便的是使用hash表。
这道题目是两数之和的延伸,基本思路和两数之和类似,先对数组进行排序,复杂度为O(nlogn)。
从小到大,先选取小的元素,再对它右边的元素用双指针法寻找。
一共会选取n-2个最小的元素,每个都对右边的元素进行双指针寻找,复杂度 为O(n^2)
稍微需要费心的是如何去除重复解,我的做法是在选取每一个最小元素之前,都先判断该元素是否
被选取过。除此之外,在添加一个三元组之前,判断第二个元素和第三个元素是否不同。如果有不同,就可以添加。
附代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int array_size=nums.size();
sort(nums.begin(),nums.begin()+array_size);
vector<vector<int>> a;
int first=0;
while(first<array_size-2&&nums[first]<=0)
{
int second=first+1,third=array_size-1;
while(second<third)
{
if(nums[second]+nums[third]>-nums[first])
third--;
else if(nums[second]+nums[third]<-nums[first])
second++;
else if(a.size()==0||(a[a.size()-1][1]!=nums[second]||a[a.size()-1][2]!=nums[third]))
{
vector<int> b={nums[first],nums[second],nums[third]};
a.push_back(b);
third--;second++;
}
else{third--;second++;}
}
do
first++;
while(nums[first-1]==nums[first]&&first<array_size-1);
}
return a;
}
};