该篇是个人对于 leetcode 15. 三数之和 解题思路
解题思路
分析
- 看起来,类似两数之和,无非是将数量增加了一个,但同时麻烦也增加了不少。
- 若是使用三层嵌套循环解题,虽然可以解决问题,但必然会导致程序超时(可行,但不可用),性能太差。
- 因为不是返回对应的索引,所以可以修改数组元素的位置,即可以排序。
步骤
- 对
nums
进行排序(递增)。 - 遍历数组,并以当前遍历位置作为分割线,在右侧数组当中(不包括分割元素在内),寻找两个可以组成
0 - nums[i]
的值,将该题间接性转换为两数之和。
事实上,该目标与 剑指 Offer 57. 和为 s 的两个数字 题意一致
优化
- 当
nums[i] > 0
时,其后续的数值都比nums[i]
大,那么就不可能存在两个数值一起组合为 0,可以提前结束遍历(前提:数组有序)。 - 若当前遍历数值与上一个数值一致(
nums[i] == nums[i - 1]
),可直接跳过(去重复)。 - 相比两数之和,与其不同的是:目标数组是有序的(
虽然是手动排序)。可使用二分查找或头尾指针等手段快速搜索目标。
重复问题
最简易的方式便是使用哈希表。
不过对于此题而言,还存在一种巧妙的去重法:双指针。
将思路拉到步骤 2,能够使用头尾指针(双指针)搜索目标是因为数组是有序的,当移动指针时,对于数值(两个指针所指向元素的和)的变化可预测的。
此处头尾指针使用
l
与r
表示。
在通过不断调整,找到目标值之后,l
与 r
都需要进行移动,并且是移动到不等于组合时的值。如 nums[l] == 0
,那么 l
需要移动至 nums[l] != 0
的位置,r
同理。
不同于两数之和,此题需要找到所有的组合可能,所以找到一个目标后,还需继续寻找其他的可能,而不能就此退出。
为什么要同时移动两个指针,不会导致错过答案吗?
并不会,比如,一个符合题意的组合 [-1, 0, 1]
,当 l
移动到了非 0 的位置时(递增变化),那么在 nums[r] = 1
的情况下,不可能再组成一个不重复的答案,组合是唯一的。
需注意,该技巧需要与第二条优化一起使用,才能达到去重的效果。
参考代码
TypeScript
function threeSum(nums: number[]): number[][] {
nums.sort((a, b) => a - b);
const res = [];
const n = nums.length;
for (let i = 0; i < n - 2; i++) {
if (nums[i] > 0) {
break;
}
const target = 0 - nums[i];
let l = i + 1;
let r = n - 1;
while (l < r) {
if (nums[l] + nums[r] === target) {
res.push([nums[i], nums[l], nums[r]]);
l++;
r--;
while (nums[l] === nums[l - 1]) {
l++;
}
while (nums[r] === nums[r + 1]) {
r--;
}
} else if (nums[l] + nums[r] < target) {
l++;
} else {
r--;
}
}
while (nums[i] === nums[i + 1]) {
i++;
}
}
return res;
}
Rust
use std::cmp::Ordering;
impl Solution {
pub fn three_sum(mut nums: Vec<i32>) -> Vec<Vec<i32>> {
nums.sort();
let n = nums.len();
let mut res = vec![];
if n < 3 {
return res;
}
let mut i = 0;
while i < n - 2 && nums[i] <= 0 {
let mut l = i + 1;
let mut r = n - 1;
while l < r {
match (nums[i] + nums[l] + nums[r]).cmp(&0) {
Ordering::Less => l += 1,
Ordering::Greater => r -= 1,
Ordering::Equal => {
res.push(vec![nums[i], nums[l], nums[r]]);
l += 1;
r -= 1;
while l < n && nums[l] == nums[l - 1] {
l += 1;
}
while r > 0 && nums[r] == nums[r + 1] {
r -= 1;
}
}
}
}
i += 1;
while i < n - 2 && nums[i] == nums[i - 1] {
i += 1;
}
}
res
}
}