题目描述
给你一个包含 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
解题思路
- 通过排序使最终得到的结果不重复
- 不讨论同一个位置的数是相同的情况
- 采用双指针的方法得到符合条件的第二个数和第三个数
代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n=nums.size();
vector<vector<int>> ans;
if(n<3) return ans;
int first,second,third;
//排序
sort(nums.begin(),nums.end());
for(first=0;first<n-2;first++){
second=first+1;
third=n-1;
while(second<third){
if(nums[first]+nums[second]+nums[third]==0){
ans.push_back({nums[first],nums[second],nums[third]});
while(second<n-1&&nums[second]==nums[second+1]) second++;
second++;
while(third>0&&nums[third]==nums[third-1]) third--;
third--;
}
else if(nums[first]+nums[second]+nums[third]<0){
while(second<n-1&&nums[second]==nums[second+1]) second++;
second++;
}
else if(nums[first]+nums[second]+nums[third]>0){
while(third>0&&nums[third]==nums[third-1]) third--;
third--;
}
}
while(first<n-1&&nums[first]==nums[first+1]) first++;
}
return ans;
}
};