引言
在今天的文章中,我们将探讨一个经典的算法问题——三数之和。这个问题要求我们在一个整数数组中找到所有不重复的三元组,使得这三个数的和为零。为了有效地解决这个问题,我们将采用排序加双指针的技巧。排序可以帮助我们更容易地跳过重复的元素,而双指针则可以在排序后的数组上高效地查找满足条件的两个数。
leetcode-三数之和
问题描述
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
解题思路
排序+双指针,演算过程如下图所示
①排序:首先对数组进行排序,这样相同的数会被放在一起,便于去重。
② 固定指针:使用一个指针 i 从数组的开始遍历到某个位置(当 nums[i] > 0 时停止,因为三个正数的和不可能为0)。
③ 双指针:在固定了 nums[i] 之后,使用两个指针 left 和 right 分别在 i 的右侧和数组的末尾进行移动,寻找满足 nums[i] + nums[left] + nums[right] == 0 的 left 和 right。
④去重: 在遍历 i 时,如果 nums[i] 与前一个元素相同,则跳过,避免重复的三元组。 在移动 left 和 right 时,如果 nums[left] 或 nums[right] 与前一个元素相同,则继续移动指针,避免重复的三元组。
⑤ 记录结果:当找到满足条件的三元组时,将其添加到结果列表中。
代码实现
以下是使用C++实现的代码示例:
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ret;
int n = nums.size(), left = 0, right = n-1;
sort(nums.begin(),nums.end());
for(int i=0; i<n; ){
if(nums[i] > 0 ) break;
left = i+1;
right = n-1;
while(left < right){
int sum = nums[left] + nums[right];
if(sum < -nums[i]) left++;
else if(sum > -nums[i]) right--;
else{
ret.push_back({nums[i],nums[left],nums[right]});
left++,right--;
while(nums[left] == nums[left-1] && left < right) left++;
while(nums[right] == nums[right+1] && left < right) right--;
}
}
i++;
while(i<n && (nums[i] == nums[i-1]) ) i++; //注:必须i<n在第二个条件前,否则数组越界
}
return ret;
}
};
这段代码首先对数组进行排序,然后使用三层循环(一层固定指针,两层双指针)来寻找满足条件的三元组,并在过程中进行去重。最终返回所有满足条件的三元组列表。时间复杂度是:O(n^2)。