双指针法
首先对数组进行排序,排序后固定一个数 nums[i],再使用左右指针指向nums[i]后面的两端,数字分别为nums[L]和nums[R],计算三个数的和 sum 判断是否满足为0,满足则添加进结果集
以下是需要注意的特殊情况:
1、如果nums[i]大于0,则三数之和必然无法等于0,结束循环,这里用break是因为nums[i]大于0说明了>i的全部值其三数之和都会大于0,没有再循环的必要,所以直接退出循环
2、如果nums[i] = nums[i−1],则说明该数字重复,会导致结果重复,所以应该跳过,这里用continue是因为只需要跳过本轮循环
3、当sum=0时,nums[L] = nums[L+1] 则会导致结果重复,应该跳过,L++
4、同理,当sum=0时,nums[R] = nums[R-1] 则会导致结果重复,应该跳过,R–
上面第3、4步的去重要用while而不是if,因为有可能不止重复一次
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end()); //排序
int len=nums.size();
vector<vector<int>> ans;
int l,r;
for(int i=0;i<len;i++){
if(nums[i]>0){ // 如果当前数字大于0,则三数之和一定大于0,所以结束循环
break;
}
if(i > 0 && nums[i] == nums[i-1]) continue; // 去重,一样的i则跳过
l=i+1; //l指向i后面的第一个
r=len-1; //r指向数组的最后一个
while(l<r){
int sum=nums[i]+nums[l]+nums[r];
if(sum==0){
ans.push_back({nums[i],nums[l],nums[r]});
//将这三个值以对象形式放入数组中
while(l<r&&nums[l]==nums[l+1]) l++;
//去重,假如下一个l和这当前l一样则跳过
while(l<r&&nums[r]==nums[r-1]) r--; //同理
l++; //左右指针向中间移动
r--;
}
else if(sum<0) l++; //sum<0,所以要较小值变大,即l右移
else if(sum>0) r--; //sum>0,所以要较大值变小,即r右移
}
}
return ans;
}
};