一、题目描述:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、思路:
首先把nums数组排好序,然后使用双指针法进行判断是否能找到三数之和等于0的情况,遍历nums数组,i表示当前遍历的元素,left指针指向i+1个元素,right指向最后一个元素,判断这三个元素相加是否等于0即可,最要注意的细节是去重,直接看代码吧。
三、代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
//排序
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++){
//如果第一个数已经大于0了,后面的就不用看了,直接返回结果
if(nums[i]>0){
return res;
}
//定义两个指针
int left=i+1;
int right=nums.size()-1;
//去重,从i之后找,如果第i个已经找过
//错误情况,[-1,-1,2]忽略了这种情况
// if(i<nums.size() && nums[i]==nums[i+1]){
// continue;
// }
if(i>0 && nums[i]==nums[i-1]){
continue;
}
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({nums[i], nums[left], nums[right]});
//去重
//如果right下标对应的值已经用到过了,right左移
while(left<right && nums[right]==nums[right-1]){
right--;
}
//同理,右移
while(left<right && nums[left]==nums[left+1]){
left++;
}
//找到符合条件的后指针同时收缩
right--;
left++;
}
}
}
return res;
}
};
最主要的细节其实还是去重部分。