给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:排序+双指针;
先将数组从小到大排序,从i=0开始,L指针指向i+1,R指针指向l-1(l为数组长度),判断nums[L]+nums[R]是否等于-nums[i]. 如果大于,则R指针减一,小于则L指针加一,相等就存于set中。set用于去重。
代码:(很慢)
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
int l=nums.size();
set<vector<int>>se;
for(int i=0;i<l-2;i++)
{
if(nums[i]>0)
break;
int target=-nums[i];
int j=i+1;
int k=l-1;
while(j<k)
{
auto sum=nums[j]+nums[k];
if(target==sum)
{
se.insert({nums[i],nums[j],nums[k]});
j++;
k--;
}
else if(sum>target)
{
k--;
}
else if(sum<target)
{
j++;
}
}
}
return vector<vector<int>>(se.begin(), se.end());
}
};