例题 :给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
双指针解法
这道题目使用哈希法并不十分合适,因为在去重的操作中有很多细节需要注意,在面试中很难直接写出没有bug的代码。而且使用哈希法 在使用两层for循环的时候,能做的剪枝操作很有限。接下来我来介绍另一个解法:双指针法,这道题目使用双指针法 要比哈希法高效一些,那么来讲解一下具体实现的思路。动画效果如下
拿这个nums数组来举例,首先将数组排序,然后有一层for循环,i从下标0的地方开始,同时定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上。依然还是在数组中找到 abc 使得a + b +c =0,我们这里相当于 a = nums[i] b = nums[left] c = nums[right]。接下来如何移动left 和right呢, 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了,left 就向右移动,才能让三数之和大一些,直到left与right相遇为止。
代码如下
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>>res;
int n=nums.size();
sort(nums.begin(),nums.end());//先对数组进行排序
if(nums.size()<3)return {};
for(int i=0;i<n;i++){
if(nums[i]>0) break;//排好序的数组如果第一个数大于0,那么后面也一定大0,这样三数之和就不可能等于0
//对第一个数去重,但是不可以写成if(nums[i]==nums[i+1]) continue; 这样的话就会忽略[-1,-1,2]这种情况,所以是在i>0的前提下去重,即遍历一遍是否存在上述情况之后去重
if(i>0&&nums[i]==nums[i-1]){
continue;
}
int l=i+1;
int r=n-1;
while(l<r){
if(nums[i]+nums[l]+nums[r]<0){
l++;//两数之和太小,左指针右移
}
else if(nums[i]+nums[l]+nums[r]>0){
r--;//两数之和太大,右指针左移
}
else{
//找到和为0的三元组,添加到结果集
res.push_back(vector<int>{nums[i],nums[l],nums[r]});
l++;//左右指针内缩,继续寻找
r--;
//第二个数和第三个数去重
while(l<r&&nums[r]==nums[r+1]) r--;
while(l<r&&nums[l]==nums[l-1]) l++;
}
}
}
return res;
}
};