给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
方法一:曾设想使用三层循环来暴力求解,但发现时间复杂度过高并且要求不重复的三元组还需要使用哈希,代码也复杂。 优化后可以使用两层循环,第二层循环加双指针,可以更好的完成
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ans;
int n = nums.size();
if (n <3){
return ans;
}
sort(nums.begin(),nums.end());
for(int first = 0;first<n;first++){
if(first >0 && nums[first] == nums[first-1]){
continue;
}
int third = n-1;
int target = -nums[first];
for(int second = first +1 ;second <n;second++){
if(second >first+1 && nums[second] == nums[second-1]){
continue;
}
while(second<third && nums[second]+nums[third] >target){
third --;}
if(third == second){
break;
}
if(nums[second]+nums[third] == target){
ans.push_back({nums[first],nums[second],nums[third]});
}
}
}
return ans;
}
};