问题描述
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题思路
- 第一反应不会做,看了题解不难。
- 对撞指针 + 排序 + 剪枝优化
- a + b + c = 0 -> 固定 nums[i] = fix,找 nums[i] + nums[r] = -fix
- 执行时间:112 ms,超越97.54%的用户
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if(nums.size()<3){
return {};
}
vector<vector<int> > res;
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size()-2;i++){
if(i>0&&nums[i]==nums[i-1]) continue;
int p = nums[i];
if(p>0){
break;
}
int l = i+1;
int r = nums.size()-1;
while(r>l){
if(nums[l]+nums[r]==-1*p){
if(l==i+1||r==nums.size()-1){
res.push_back(vector<int>{p,nums[l],nums[r]});
l++;
r--;
}else if(nums[l]==nums[l-1]){
l++;
}else if(nums[r]==nums[r+1]){
r--;
}else{
res.push_back(vector<int>{p,nums[l],nums[r]});
l++;
r--;
}
}else{
if(nums[l]+nums[r]>-p){
r--;
}else{
l++;
}
}
}
}
return res;
}
};