给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
这道题其实不难,一定要先排序,固定两个指针,移动另外一个,麻烦的是重复值得处理
// O(n^2)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
std::sort(nums.begin(), nums.end());
vector<vector<int>> ans;
for(int c = nums.size() - 1; c >= 2; c--){
for(int a = 0, b = c - 1; a < b; ){
int sum = nums[a] + nums[b];
if(sum > -nums[c])
b--;
else if(sum < -nums[c])
a++;
else{
vector<int> nums_result;
nums_result.push_back(nums[a]);
nums_result.push_back(nums[b]);
nums_result.push_back(nums[c]);
ans.push_back(nums_result);
while(a<b&&nums[a] == nums[a+1]) a++;
a++;
}
}
while(c >= 1 && nums[c] == nums[c-1]) c--;
}
return ans;
}
};