题目
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
首先将数组进行排序,因为需要的是三数和为0,所以先选择一个最小的数nums[0],因为是不需要相同的三元组和,所以nums[i]!=nums[i-1],假如相等就直接跳过,但是有一个前提就是i不能等于0,因为如果在i等于0的时候也进行nums[i]!=nums[i-1]的判断,那么在[0,0,0,0]这种样本就不能通过,因为i等于0的时候因为是第一次进行三数之和等于0的判断,结果数组中目前没有任何有效的结果,肯定不能重复,所以不需要再做nums[i]!=nums[i-1]的判断,选定i之后,因为要等于0,首先选择i后面元素中的最大值和最小值,也就是nums[i+1] and nums[len-1],再将三者的和与0做比较,大于0则代表大的那个值取得过大,需要进行索引减1操作,小于0代表小的那个值过小,将其索引加一,在这个过程中必须保证l<r,最后得到等于0的结果旧作为结果保存起来,然后防止重复值,然后i再往前加,直到加到i等于len-2为止。
code
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if(nums.size()<3)
return {};
sort(nums.begin(),nums.end());
vector<vector<int>>res;
for(int i=0;i<nums.size()-2;++i)
{
int l=i+1;
int r=nums.size()-1;
while(l<r)
{
if(i==0||nums[i]!=nums[i-1])
{
int target=nums[i]+nums[l]+nums[r];
if(target==0)
{
res.push_back({nums[i],nums[l],nums[r]});
l++;
r--;
while(l<r&&nums[l]==nums[l-1])
l++;
while(l<r&&nums[r]==nums[r+1])
r--;
}
else if(target<0)
{
l++;
}
else
{
r--;
}
}
else
break;
}
}
return res;
}
};