15. 三数之和
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
链接:https://leetcode-cn.com/problems/3sum
示例:给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
算法:将三数和转化为两数和定于第三数的相反数;
1.将数组进行升序排列;
2.若数组长度小于三 or 数组front元素大于零 or 数组back元素小于零 ,则返回空 ;
3.遍历数组,将所遍历的元素取反,视为“第三个数的相反数”:
a.若该元素大于零,终止循环;
b.若该元素等于上一个元素,跳出本层循环,继续执行;
b.设置两外两个元素的下标指针 h=i+1, r=len-1 :
。若三元素之和等于零,将三元素加入结果序列 ,将更新h元素和r元素(后续元素不能与当前元素相同,防止出现重复元素),h++, r--;
。若三元素之和大于零,r-- ;
。若三元素之和小于零,h++ ;
4.返回结果序列;
代码:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res ;
int len = nums.size() ;
sort(nums.begin(), nums.end()) ;
if (len<3 || nums.front()>0 || nums.back()<0)
return {} ;
for (int i = 0 ; i < len - 2 ; i++){
if (i>0 && nums[i] == nums[i-1])
continue ;
int h = i + 1 ;
int r = len -1 ;
while (r > h){
if (nums[i]>0)
break ;
if (nums[i] + nums[h] + nums[r] == 0 ){
res.push_back({nums[i], nums[h], nums[r]});
while (h < r && nums[h] == nums[h+1])
h++ ;
while (h < r && nums[r] == nums[r-1])
r-- ;
h++ ;
r-- ;
}
else if (nums[i] + nums[h] + nums[r] > 0){
r-- ;
}
else if (nums[i] + nums[h] + nums[r] < 0)
h++ ;
}
}
return res ;
}
};