题目描述:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解题思路:
先用sort函数将原数组按从小到大顺序排列,方便后续指针自增自减的运行。因为如果数组有序,就可用双指针以线性时间复杂度来遍历数组。
先指定一个数,fix。然后看剩下的数中有没有两个数相加等于fix的相反数。
如果输入数组为空,或第一个数>0或最后一个数<0,那就输出为空。因为第一个数大于0,意味着整个数组全都大于0,不可能出现相加=0的情况;最后一个数<0也是同理。
然后开始遍历。当遍历到正数的时候就break,因为数组有序,当fix为正的时候,说明它后面的数都是正的,永远不会出现和为0的情况。还要加上重复跳过的步骤。即当前fix与前一个fix相同的时候,就跳过该数,因为同一个数遍历2次浪费时间。对于当前遍历到的fix,用0减去fix得到target,然后只需要找到2个数之和等于target即可。用2个指针分别指向fix数字之后开始的数组首尾两个数,如果两个数和正好为target,则将这两个数和fix的数一起存入结果中。然后进行指针跳过重复数值的步骤。如果两数之和小于target,则我们将左边那个指针i右移一位,使得指向的数字增大一些。同理,如果两数之和大于target,则我们将右边那个指针j左移一位,使得指向的数字减小一些,代码如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
sort(nums.begin(),nums.end());
if(nums.size()<3 || nums.front()>0 || nums.back()<0)
return {};
for(int k=0;k<nums.size();k++)
{
if(nums[k]>0)
break;
if(k>0 && nums[k]==nums[k-1])
continue;
int target=0-nums[k];
int i=k+1;
int j=nums.size()-1;
while(i<j)
{
if(nums[i]+nums[j]==target)
{
res.push_back({nums[k],nums[i],nums[j]});
while(i<j && nums[i]==nums[i+1])
i++;
while(i<j && nums[j]==nums[j-1])
j--;
i++;
j--;
}
else if(nums[i]+nums[j]<target)
i++;
else
j--;
}
}
return res;
}
};