个人主页:兜里有颗棉花糖
欢迎 点赞 收藏✨ 留言✉ 加关注本文由 兜里有颗棉花糖 原创
收录于专栏【手撕算法系列专栏】【LeetCode】
本专栏旨在提高自己算法能力的同时,记录一下自己的学习过程,希望对大家有所帮助
希望我们一起努力、成长,共同进步。
点击直接跳转到该题目
给定一个包含 n
个整数的数组 nums
,判断 nums
中是否存在三个元素 a ,b ,c
,使得 a + b + c = 0 ?
请找出所有和为 0
且 不重复 的三元组。
示例1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例2:
输入:nums = []
输出:[]
示例3:
输入:nums = [0]
输出:[]
注意:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
本题目可以使用双指针和单调性(排序)
的思路来进行操作。具体思路如下:
双指针指向的数的和
和target
的大小关系进行双指针的移动,如果和等于target,则找到了一个满足条件的三元组,将其添加到结果数组ret中,并同时移动左指针l向右和右指针r向左。在移动指针之后,为了避免重复的结果,需要跳过相邻的相同数。具体做法是,如果左指针l指向的数与前一个数相同,就继续向右移动指针,直到找到一个不同的数为止。同样的,在移动右指针r之后,如果右指针r指向的数与后一个数相同,就继续向左移动指针,直到找到一个不同的数为止。需要注意的是:一定要注意双指针交错的情况
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<vector<int>> ret;
int n = nums.size();
for(int i = 0;i < n;)
{
int l = i + 1, r = n - 1, target = -nums[i];
while(l < r)
{
if(nums[l] + nums[r] > target) r--;
else if(nums[l] + nums[r] < target) l++;
else
{
ret.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--;
}
}
i++;
while(i < n && nums[i] == nums[i - 1]) i++;
}
return ret;
}
};
通过啦!!!