题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[ [-1, 0, 1] , [-1, -1, 2] ]
题解:
对于给到的nums数组,我们先进行一次sort排序,这样会方便我们去重,三元组的从小到大排序以及减少遍历次数
排序完成后,我们遍历nums数组,nums [ i ] 就是三元组的最左边的元素。
由于我们事先排序,所以剩下的两个元素一定在nums [ i ] 的右边,然后我们定义一对左右边界
Left=i+1;
Right=nums.size()-1;
如果 nums[ i ] + nums[ Left ]+nums[ Right ] 大于 0,我们向左移动右边界
如果 nums[ i ] + nums[ Left ]+nums[ Right ] 小于 0,我们向右移动左边界
当nums[ i ] + nums[ Left ]+nums[ Right ] 等于 0,就把这个三元组放进vector容器中,并且左边界或者右边界旁有相同的元素要进行去重,这样可以保证下一次的三元组不会重复
最后不要忘记给最左边元素去重,如果最左边元素重复,有可能会有相同的三元组被放入vector容器
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n=nums.size();
vector<vector<int>>num;
sort(nums.begin(),nums.end());
for(int i=0;i<n;i++){
if(nums[i]>0)break;//优化,因为最左边的元素已经大于0了,无论后面两个元素是什么,结果必大于0
if(i>0&&nums[i]==nums[i-1])continue;//去重
int l=i+1,r=n-1;
while(l<r){
int sum=nums[i]+nums[l]+nums[r];
if(sum==0){
num.push_back({nums[i],nums[l],nums[r]});
while(l<r&&nums[l]==nums[l+1])l++;//去重
while(l<r&&nums[r]==nums[r-1])r--;//去重
l++;r--;
}
else if (sum < 0) l++;
else if (sum > 0) r--;
}
}
return num;
}
};