具体步骤如下:
1、将数组 nums 排序,这样方便我们去重和调整双指针。
2、枚举数组中的每一个数 nums[i] 作为三元组的第一个数,将问题转化为在剩余部分 nums[j…n-1] 中找到两个数的和等于 -nums[i]。其中 n 是数组长度。
3、我们用双指针技巧,初始化左指针为 j=i+1,右指针为 k=n-1,当 j<k 的时候执行以下循环:
a:如果 nums[i]+nums[j]+nums[k]==0,则将三元组 [nums[i], nums[j], nums[k]] 添加到答案中,并同时将 j 和 k 移动到下一位置。
b:如果 nums[i]+nums[j]+nums[k]<0,则将左指针 j 向右移动一位,使得 nums[j] 变大。
c:如果 nums[i]+nums[j]+nums[k]>0,则将右指针 k 向左移动一位,使得 nums[k] 变小。
去重:如果 nums[i]=nums[i-1],则三元组中第一个数字已经出现过,为了避免重复,需要跳过当前的循环。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
//保存结果
vector<vector<int>> ans;
int n=nums.size();
//进行排序
sort(nums.begin(),nums.end());
for(int i=0;i<n;i++)
{
if(i>0&&nums[i]==nums[i-1])continue;
int j=i+1,k=n-1;
while(j<k)
{
if(nums[i]+nums[j]+nums[k]==0)
{
ans.push_back({nums[i],nums[j],nums[k]});
j++;
k--;
while(j<k&&nums[j]==nums[j-1])
j++;
while(j<k&&nums[k+1]==nums[k])
k--;
}
else if(nums[i]+nums[j]+nums[k]<0)
j++;
else
k--;
}
}
return ans;
}
};