给你一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:先进行排序,依此遍历vector,求剩下的两个值的组合,方法是定义两个索引,一个为当前索引值+1,另一个为len-1,
若nums[left]+nums[right]<1-nums[i],left++;若nums[left]+nums[right]>1-nums[i],right--;需要注意的是,遍历过程中需要判断nums[i]是否等于nums[i-1],若等于则表明有重复,需要跳过,对于右边来说就是判断nums[right]==nums[right+1]
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
vector<int> temp(3);
sort(nums.begin(),nums.end());
int len = nums.size();
for(int i=0;i<len-2;i++){
if(i>0 && nums[i]==nums[i-1])
continue;
int rest = 0-nums[i];
int left = i+1;
int right = len-1;
while(left<right){
if(nums[left]+nums[right]==rest){
temp[0]=nums[i];
temp[1]=nums[left];
temp[2]=nums[right];
result.push_back(temp);
left++;
right--;
while (left<right && nums[left] == nums[left-1])
left++;
while (left<right && nums[right] == nums[right+1])
right--;
}
else if(nums[left]+nums[right]<rest){
left++;
}
else{
right--;
}
}
}
return result;
}
};