给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
思路:排序+双指针
先固定一个元素,再通过左右指针找到两个符合的元素
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int m=nums.size();
sort(nums.begin(),nums.end()); //排序
vector<vector<int>> res;
for(int i=0;i<m;i++){
if(nums[i]>0){//第一个元素大于0,则没有结果
return res;
}
if(i>0&&nums[i]==nums[i-1]){//去除重复元素
continue;
}
int left=i+1;//左右指针
int right=m-1;
while(left<right){
if(nums[i]+nums[left]+nums[right]>0) right--;//根据大小进行左右调整
else if(nums[i]+nums[left]+nums[right]<0) left++;
else{
res.push_back(vector<int>{nums[i],nums[left],nums[right]});//找到结果后排入返回容器
left++;//收缩左右指针
right--;
while(left<right&&nums[left]==nums[left-1]) left++;//去重
while(left<right&&nums[right]==nums[right+1]) right--;
}
}
}
return res;
}
};