文章目录
题目描述
题目链接:15.三数之和
算法原理(排序+双指针)
本题与两数之和类似,是⾮常经典的⾯试题。与两数之和稍微不同的是,题⽬中要求找到所有不重复的三元组。那我们可以利⽤在两数之和那⾥⽤的双指针思想,来对我们的暴⼒枚举做优化:
- 先排序;
- 然后固定⼀个数a
- 在这个数后⾯的区间内,使⽤双指针算法快速找到两个数之和等于-a 即可。
但是要注意的是,这道题⾥⾯需要有去重操作~
- 找到⼀个结果之后, left 和right 指针要跳过重复的元素;
- 当使⽤完⼀次双指针算法之后,固定的a 也要跳过重复的元素。
代码实现
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> vv;
sort(nums.begin(),nums.end());//1.排序
for(int i = 0;i < nums.size() - 1;)//2.固定一个数
{
if(nums[i] > 0)break;//小优化
int left = i + 1,right = nums.size() - 1,target = -nums[i];
while(left < right)//3.利用双指针算法快速找出等于|nums[i]|的组合
{
int sum = nums[left] + nums[right];
if(sum > target)--right;
else if(sum <target)++left;
else
{
vv.push_back({nums[i],nums[left],nums[right]});
++left,--right;
while(left < right&& nums[left] == nums[left - 1])++left;//去重left
while(left < right&& nums[right] == nums[right + 1])--right;//去重right
}
}
++i;
while(nums[i] == nums[i - 1]&&i < nums.size() - 1)++i;//去重
}
return vv;
}
};