15.三数之和
题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
来源:力扣(LeetCode)
链接:题目来源
示例1
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例2
输入:nums = []
输出:[]
示例3
输入:nums = [0]
输出:[]
思路
1)开始分析的时候,一开始想到的是先把数组进行排序,然后双层循环控制前两个数,再寻找第三个数等于0减去前两个数之和。但是因为每次寻找是都需要从第二个数之后开始找,这样的话当数据量很大的时候直接超时。
2)看了大佬的代码以后,把我的三部分变成三个指针来分别指向对应的点,先给数组从小到大排序,第一个first指针在for循环中从左扫描到右,第二个指针在第二层for循环中从左扫到右,每当第一个和第二个指针确定时,我们就用第三个指针从右往左扫描数组直到找到合适的值使得他们三个之和为0。
代码
vector<vector<int>> threeSum(vector<int>& nums)
{
int n=nums.size(); //数组长度
sort(nums.begin(),nums.end()); //从小到大排序
vector<vector<int>>num; //输出数组
if(n<3 || nums[n-1]<0) return num; //当数组中的数字数量小于三个或者最大值小于0时,直接输出空数组
for (int first=0;first<n-1;first++)
{
//确定第一个指针指向的数与之前不同,防止产生重复的数组
if(first>0 && nums[first]==nums[first-1]) continue;
//第三个指针开始指向最后一个数
int third = n-1;
//找到目标值为第二个数与第三个数之和应该等于的数
int target = -nums[first];
for(int second=first+1;second<n;second++)
{
//确定第二个指针指向的数与之前不同,防止产生重复的数组
if(second>first+1 && nums[second]==nums[second-1]) continue;
//第三个指针往前扫描,寻找合适的数与第二个数之和为目标值
while(second<third && nums[second]+nums[third]>target) third--;
//如果他们相遇,已经不会产生合适的且不重复的解,直接退出该循环
if(third==second) break;
//找到合适的数加入目标数组
if(nums[second]+nums[third]==target)
{
num.push_back({nums[first],nums[second],nums[third]});
}
}
}
return num;
}