题目描述
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
代码示例
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
std::sort(nums.begin(), nums.end());
vector<vector<int>> v2;
//a+b+c = 0 --> a+b = -c
for(int c = nums.size()-1;c >=2; ){
for(int a = 0,b = c-1;a < b; ){
int sum = nums[a] + nums[b];
if(sum < -nums[c]){
++a;
}
else if(sum > -nums[c]){
--b;
}
else{
vector<int> v = {nums[a], nums[b], nums[c]};
v2.push_back(v);
//去重复 a b
do{
++a;
} while(a < b && nums[a-1] == nums[a]);
do{
--b;
} while(a < b && nums[b+1] == nums[b]);
}
}
do{
--c;
} while(c >= 2 && nums[c+1] == nums[c]);
}
return v2;
}
};
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int target;
vector<vector<int>> ans;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++) {
if (i > 0 && nums[i] == nums[i - 1]) continue;
if ((target = nums[i]) > 0) break;
int l = i + 1, r = nums.size() - 1;
while (l < r) {
if (nums[l] + nums[r] + target < 0) ++l;
else if (nums[l] + nums[r] + target > 0) --r;
else {
ans.push_back({target, nums[l], nums[r]});
++l, --r;
while (l < r && nums[l] == nums[l - 1]) ++l;
while (l < r && nums[r] == nums[r + 1]) --r;
}
}
}
return ans;
}
};