题目描述
给定一个包含 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]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
解题思路
首先给数组排序,然后遍历数组,对每个nums[i],用双指针去寻找另外两个和为0-nums[i]的数,注意去重
代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n=nums.size();
if(n<3) return {};
sort(nums.begin(),nums.end());
vector<vector<int>>ans;
int i,a,b;
for(i=0;i<n;i++){
//去重
if(i>0&&nums[i]==nums[i-1]) continue;
int temp=0-nums[i];
a=i+1;b=n-1;
while(a<b){
if(nums[a]+nums[b]==temp){
//去重
if((a-1>i&&nums[a]==nums[a-1])||(b+1<n&&nums[b]==nums[b+1])) {
a++;
b--;
continue;
}
ans.push_back({nums[i],nums[a],nums[b]});
a++;
b--;
continue;
}
else if(nums[a]+nums[b]>temp){
b--;
}
else if(nums[a]+nums[b]<temp){
a++;
}
}
}
return ans;
}
};