题目来源
解题方法
排序+双指针
三个变量i,j,k分别表示三个数的下标
题目要求需要无重复的答案,我们就可以考虑先排序,因为排序后的数组后一个总是大于等于当前元素,如果相等的话我们就继续往后一个即可
双指针思路的来源:
我们可以发现在确定第一个数的情况下,第二个数的大小与第三个数是成反比的,如果三数之和大于0,说明第三个数大了,那就可以让第三个数变小,即k- -,如果三数之和等于0,就存入res
我们选择在第一个数更新的时候更新k,原因是第二个数更新的时候,只会越变越大,那么对应的第三个数只会越来越小
每次i和j更新的时候要保证与上一次的数不一样
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.size()<3)
return res;
int n=nums.size();
sort(nums.begin(),nums.end());
for(int i=0;i<n-2;i++){
int k=n-1;
if(i>0&&nums[i]==nums[i-1])
continue;
for(int j=i+1;j<n;j++){
if(j>i+1&&nums[j]==nums[j-1])
continue;
while(j<k&&nums[i]+nums[j]+nums[k]>0)
k--;
if(j==k)
break;
if(nums[i]+nums[j]+nums[k]==0)
res.push_back({nums[i],nums[j],nums[k]});
}
}
return res;
}
};