题目如下所示:
一开始我的想法是先对数组进行排序,然后针对其中数组大小小于3的特例返回空,寻找三个元素的方法是通过两个for循环,首先设置i和j,计算两个位置对应元素之和,然后在数组中用find函数,寻找它的相反数,这种方法还没有考虑到重复性的问题,所以会导致所求解出来的结果有重复。所以我加上了如果第i个与第i+1相同则跳过的程序,但是会导致有些符合条件的元素没有被选中。
最终选择通过官方解法,排序加双指针。需要强调的是,以后求解需要有双指针的思维,不能总是暴力求解。
算法流程出自评论区的一个老哥:
通过它的算法流程,编写了c++程序,如下所示:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums)
{
int n=nums.size();
vector<vector<int>> res={};
if(n<3)
{
return res;
}
sort(nums.begin(),nums.end());
for(int i=0;i<nums.size();i++)
{
if(nums[i]>0)
{
return res;
}
if(i>0&&nums[i]==nums[i-1])
{
continue;
}
int L=i+1;
int R=n-1;
while(L<R)
{
if(nums[i]+nums[L]+nums[R]==0)
{
res.push_back({nums[i],nums[L],nums[R]});
while(L<R&&nums[L]==nums[L+1])
{
L=L+1;
}
while(L<R&&nums[R]==nums[R-1])
{
R=R-1;
}
L=L+1;
R=R-1;
}
else if(nums[i]+nums[L]+nums[R]>0)
{
R=R-1;
}
else
{
L=L+1;
}
}
}
return res;
}
};