给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:
之前我们已经做过《两数之和》,他是判断一个数组里是否存在2个数,这2个数的和为给定的target的值
对于三数之和,我们只需把它转化为《两数之和》的问题就可以,然后用《两数之和》的解法稍变化下就可以了
因为要求是3个数n1,n2,n3的和为0,那么转化一下,可以变成求2个数n2,n3,这2个数的和为(0-n1),这就成功转化成两数之和的问题了
做法:
先对数组进行排序,然后从第一个元素开始,每次把target置为0-nums[i],然后判断是否能在后面的数中找出2个数,这2个数的和为0-nums[i],如果可以找到,就把这3个数存到要返回的数组中
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> rev;
if(nums.empty())
return rev;
vector<int> temp;
sort(nums.begin(),nums.end()); //先对数组进行排序
int target,left,right;
for(int i = 0; i < nums.size()-1; i++)
{
if(nums[i] > 0) //如果nums[i]>0,不可能找到,直接结束循环
break;
else if(i>0 && nums[i]== nums[i-1]) //如果这个数在前面已经判断过,则直接跳过进行下一次循环
continue;
else
{
target = 0 - nums[i]; //target初始化
left= i + 1; //left初始化为后面的数的最左边的数
right = nums.size()-1; //right初始化为后面的数的最右边的数
while(left<right)
{
if(nums[left] + nums[right] > target) //和大于target的情况
right--;
else if(nums[left] + nums[right] < target) //和小于target的情况
left++;
else //和等于于target的情况
{
temp.push_back(nums[i]);
temp.push_back(nums[left]);
temp.push_back(nums[right]);
rev.push_back(temp);
temp.clear();
while(left < right && nums[left] == nums[left+1]) //要继续寻找,不能只找到一个就停,并且还要避免重复的情况
left++;
while(right > left && nums[right] == nums[right-1])
right--;
left++;
right--;
}
}
}
}
return rev;
}
};
效率:99.18%