题目
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
解题思路
本题目是也是找数字,容易想到采用哈希法,但是在使用哈希法的过程中可以发现,题目是要求答案不可以包含重复三元组的(三元组内元素是可以重复的),单纯使用哈希法我们要判断的查重条件过于复杂,一不小心就可能漏掉了某个查重条件。
为此我们引入了双指针,结合哈希法对这题目进行处理,就没有那么多的查重条件判断了
下面是具体思路:
-
定义一个二维数组用于存储返回的答案
-
对数组进行排序,便于查重
-
从前往后遍历数组,暂时把第一个遇到的数作为三元组第一数(当第一个元素大于0时返回结果,因为我们是个排序数组,易知后面不可能再出现俩数使三数和等于0)
-
对三元组第一数进行查重,注意要查已经遍历过的元素而不是将要遍历的元素,即只能是
nums[i] == nums[i - 1]
而不是(会导致漏掉-1,-1,2这种情况)
nums[i] == nums[i + 1]
-
定一个下标left 定义在i+1的位置上,定义下标right 在数组结尾的位置上;如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了,因为数组是排序后了,所以right下标就应该向左移动,这样才能让三数之和小一些。同理left;
-
找到结果后先将结果存入结果集再分别对left和right查重,最后别忘记双指针也要同时内缩一次
代码实现如下:
//双指针+哈希法
#include <vector>
#include<algorithm>
using namespace std;
class Solution
{
public:
vector<vector<int>> threeSum(vector<int>& nums)
{
vector<vector<int>> result;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size(); i++)
{
//第一个值大于零就没必要继续下去 不可能在排序情况下三数之和为0
if (nums[i] > 0)
{
return result;
}
//对第一个数进行查重,不能为nuns[i] == nums[i + 1]的形式
if (i > 0 && nums[i] == nums[i - 1])
{
continue;
}
int left = i + 1;
int right = nums.size() - 1;
while (left < right)
{
if (nums[i] + nums[left] + nums[right] > 0)right--;
else if (nums[i] + nums[left] + nums[right] < 0) left++;
else
{ //将结果存入result中
result.push_back(vector<int>{nums[i], nums[left], nums[right]});
//存结果后再对后面的可能进行查重
while (left < right && nums[left] == nums[left + 1])left++;
while (left < right && nums[right] == nums[right - 1])right--;
//找到答案双指针内缩
left++;
right--;
}
}
}
return result;
}
};
总结
有些题型虽然是找元素要用到哈希法,但还是要根据题目具体要求灵活运用其他的知识,比如这个题目当中的双指针,在前面刷题的过程中也多次遇到过,还需努力。