力扣练习第九天——三数之和
题目大致如下:
给定一个包含n个整数的数组nums,判断nums中是否存在三个元素a,b,c,使得a+b+c=0?找到所有满足条件且不重复的三元组。
注:答案中不包含重复的三元组。
示例:
给定数组nums=[-1,0,1,2,-1,-4],
则满足要求的三元组为:
[ [-1,0,1],[-1,-1,2] ]
链接来源于LeetCode:https://leetcode-cn.com/problems/3sum/
大致思路:
首先想的是利用sort函数给数组排序,但是在查找过程中,如果运用过多的循环,导致程序运行超时。所以借鉴了他们的双指针的办法找和为0的数字。当和大于或者小于0的时候,双指针交替向中间移动。
代码大致如下:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
vector<vector<int>> res;
if(nums.size() < 2) return res;
int l = 0;int r = 0;int sum = 0;
for(int i = 0; i < nums.size() - 2; i++)
{
if(nums[i] > 0) break;
if(i > 0 && nums[i] == nums[i - 1]) continue;
l = i + 1;
r = nums.size() - 1;
while(l < r){
sum = nums[i] + nums[l] + nums[r];
if(sum == 0){
res.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--;
}
else if(sum < 0) l++;else r--;
}
}
return res;
}
};
结果大致为: