https://leetcode-cn.com/problems/3sum/
一开始想到了两数之和的做法,利用map,可以在时间复杂度O(N^2)的条件下完成,但是没有写,感觉很麻烦。。。果然后来看了一些分析,判断不能出现重复项是一件很麻烦的事情。。。
这道题最大的收获感觉就是或许可以尝试使用stl中一些算法,比如sort大体应该是使用快速排序完成的,平均时间复杂度是O(nlogn),是比O(N^2)快的。
在排序后,使用双指针(三指针)。类似于两数之和中利用map的做法。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>>result;
int sz=nums.size();
sort(nums.begin(),nums.end());
for(int first=0;first<sz;++first){
if(first>0&&nums[first]==nums[first-1])
continue;//保证first不重复
int third=sz-1;
for(int second=first+1;second<sz;++second){
if(second>first+1&&nums[second]==nums[second-1])
continue;
while(second<third&&nums[second]+nums[first]+nums[third]>0){
--third;
}
if(second==third)
break;
if(nums[second]+nums[first]+nums[third]==0)
result.push_back({nums[first],nums[second],nums[third]});
}
}
return result;
}
};
时间复杂度O(N^2)