题目中说的不可以包含重复的三元组,从示例1可以看出[-1,0,1] 和[0,1,-1]虽然三个数顺序不同但是元素重复了,所以只选取其中一个。而本题难点也在于去重。
对数组排序后使用双指针,借助排序后呈现的单调性降低时间复杂度。
对于用双指针寻找一个目标和我们之前做过了,而对于三个数的和是否也适用?其实只要将三个数的其中一个数固定,而去找另两个数的和,就可以化用以前的知识了。假设这个固定的数是a,那么我们要找的和就是-a。
难点其实在于处理细节:
1 去重
找到一种符合要求的结果后,左右指针要跳过重复的数(双指针的选取的去重)
而当一次双指针寻找循环结束后,固定数移动时,如果遇到了重复的数,则也要跳过。(固定数的选取的去重)
此外,在进行去重时,还要注意不要越界。
2 避免遗漏
当找到一组三元组符合题目要求时,不能停止循环,而是继续寻找。也就是让双指针继续移动。
因为固定的数也要去重,所以在用for循环时,在for后的式子中省略了最后一个调整,把调整放进了去重的操作中。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ret;
//排序
sort(nums.begin(),nums.end());
//双指针 单调性
int n = nums.size();
for(int i = 0;i < n;)
{
if(nums[i]>0)
{
break;
}
int left = i+1;
int right = n-1;
while(left<right)
{
if(nums[left] + nums[right] == -nums[i])
{
ret.push_back({nums[i],nums[left],nums[right]});
left++;
right--;
// left right去重
while(left < right && nums[left] == nums[left-1]) left++;
while(left < right && nums[right]==nums[right+1]) right--;
}
else if(nums[left] + nums[right] < -nums[i])
{
left++;
}
else if(nums[left] + nums[right]> -nums[i])
{
right--;
}
}
//一次双指针算法结束
i++;
//i 去重
while(i < n && nums[i] == nums[i-1])
{
i++;
}
}
return ret;
}
};