给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
用双指针。先确定第一个数,然后定义一个l 和 r,l为第二个数,r为第三个数。起始时l在i + 1的位置,r在数组末尾。判断是否满足条件,再根据情况移动l和r。
因为要去重,相邻两次枚举的元素不能相同。若a[i] == a[i + 1],枚举a[i]出现的情况,a[i + 1]跳过,不能跳过a[i]而枚举a[i + 1]的情况。
class Solution {
typedef long long ll;
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int> >ans;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size(); i++) {//枚举第一个数
if(i && nums[i] == nums[i - 1]) continue;//去重
int l = i + 1, r = nums.size() - 1;
while(l < r) {
if(l > i + 1 && nums[l] == nums[l - 1]) {//去重
l++;
continue;
}
int sum = nums[i] + nums[l] + nums[r];
if(sum > 0) r--;//根据情况移动指针
else if(sum < 0) l++;
else {
ans.push_back({nums[i], nums[l], nums[r]});
l++, r--;
}
}
}
return ans;
}
};