题目描述:
此题如果纯暴力的话应该是会超时的(没试过)考虑在暴力的基础上优化一下,先对数组排序,然后枚举第一个数 i,那么剩下两个数之和为 j + k = 0 - i,然后设立两个指针j和k指向头和尾,对剩下的数组从两头到中间搜索答案,时间复杂度为o(n方),代码如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.size() < 3) return res;
sort(nums.begin(),nums.end());
int j,k,target;
for(int i = 0; i < nums.size() - 2; i++){
if(i != 0){
if(nums[i] == nums[i-1]) continue;//避免重复搜索,比如 -2,-2,0,1,2。。。。。搜完第一个-2应跳过第二个-2
}
j = i + 1;
k = nums.size() - 1;
target = 0 - nums[i];
while(j < k){
if(nums[j] + nums[k] == target){
res.push_back({nums[i],nums[j],nums[k]});
while(nums[j] == nums[j+1] && nums[k] == nums[k-1]){//这里是为了避免形如 -2,0,0,2,2这种情况
j++;
k--;
}
j++;
k--;
}
else if(nums[j] + nums[k] < target) j++;
else k--;
}
}
return vector<vector<int>>(res.begin(),res.end());
}
};