力扣题目链接:三数之和
一、题目解析
二、算法原理
解法一:排序+暴力枚举+利用set去重
代码就不写了,你们可以试着写一下
解法二:排序+双指针
这题和上一篇文章的两数字和方法类似
- 排序
- 固定一个数a
- 在这个数的后面区间,使用双指针找到两个数之和为-a即可
需要解决两个细节问题:
1. 去重(避免重复的三元组)
- 找到一种结果后left和right要跳过重复的元素
- 当双指针使用完跳出循环后,a也需要跳过重复的元素
去重的时候还要控制边界,避免越界的问题
2. 不漏(不漏掉任何一个满足条件的三元组)
找到一种结果后不要“停”(left++,right--),缩小区间,继续往后寻找。
三、代码编写
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ret;
//1.排序
sort(nums.begin(), nums.end());
int n = nums.size();
//2.双指针
for(int i = 0; i < n; )
{
if(nums[i] > 0) break;//小优化:大于0的直接退出循环不用往后找了
int left = i + 1, right = n - 1, target = -nums[i];
while(left < right)
{
int sum = nums[left] + nums[right];
if(sum > target) right--;
else if (sum < target) left++;
else
{
ret.push_back({nums[i], nums[left], nums[right]});
left++;
right--;
//去重left和right
while(left < right && nums[left] == nums[left - 1]) left++;
while(left < right && nums[right] == nums[right + 1]) right--;
}
}
i++;
//去重i
while(i < n && nums[i] == nums[i - 1]) i++;
}
return ret;
}
};
这里还有个小优化,当a大于0的话可以不用在往后找,因为后面都是大于0的数,再怎么找也不会满足条件的。