本题要求读者先掌握 两数之和 II - 输入有序数组 这道题,最好看我的 个人题解 以便于衔接本篇思路。
思路
先对数组排序,设一非递减的数组示例和初始三指针位置及名字如下所示。
固定
i
i
i,即可转换为寻找满足
n
u
m
s
[
l
]
+
n
u
m
s
[
r
]
=
−
n
u
m
s
[
i
]
nums[l] + nums[r] = -nums[i]
nums[l]+nums[r]=−nums[i] 的三元组,借用两数之和的思路即可。
还要注意因为不能包含重复的三元组,以下两个三元组只能取一个,而后我们再考虑其是否满足 n u m s [ l ] + n u m s [ r ] = − n u m s [ i ] nums[l] + nums[r] = -nums[i] nums[l]+nums[r]=−nums[i]。
所以在移动指针的时候,需要规避连续的重复元素,比如下方的移动
r
r
r,对于
i
,
l
i, l
i,l 同理。
题目中要求不重复,所以从小到大枚举 i i i, l l l 的初值为 i + 1 i+1 i+1, r r r 的初值一直为 n u m s . s i z e − 1 nums.size - 1 nums.size−1。
下图为
i
=
1
i = 1
i=1 的三指针初始位置。
代码
仅提供 java 代码,多语言代码块可看这里。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
// 待返回的三元组
List<List<Integer>> triples = new ArrayList<>();
for (int i = 0; i < nums.length; i++) {
// 检测重复的 nums[i]
if (i > 0 && nums[i] == nums[i - 1]) continue;
int l = i + 1;
int r = nums.length - 1;
while (l < r) {
// 检测重复的 nums[l]
if (l > i + 1 && nums[l] == nums[l - 1]) {
l++;
}
// 检测重复的 nums[r]
else if (r < nums.length - 1 && nums[r] == nums[r + 1]) {
r--;
}
// 均不重复再按照两数之和的思路
// 等同于判断 nums[l] + nums[r] > -nums[i]
else if (nums[i] + nums[l] + nums[r] > 0) {
r--;
}
// 等同于判断 nums[l] + nums[r] < -nums[i]
else if (nums[i] + nums[l] + nums[r] < 0) {
l++;
} else { // nums[l] + nums[r] + nums[i] == 0, 三元组符合,添加入结果
triples.add(Arrays.asList(nums[i], nums[l], nums[r]));
l++;
r--;
}
}
}
return triples;
}
}
笔者代码中少了一层嵌套循环(存疑可看这里的详细对比),但复杂度不变。虽然耗时明显更慢,但在面试中是肯定能通过的。此外这种技巧在实际开发中也能用到,而且实际开发的内部操作繁多,相比之下,这种减少嵌套循环增加的耗时就可以忽略了。
复杂度
时间:
O
(
n
2
)
O(n^2)
O(n2)
空间:
Θ
(
l
o
g
n
)
\Theta(log\ n)
Θ(log n),主要考虑排序,忽略返回值。
推广
这里有附个人题解的双指针题单。
笔者著有 Leetbooks 《图论入门》,《图论进阶》,且在陆续写其他 Leetbooks。致力于提升广大读者真正的算法水平,并兼顾不同阶段的读者,如本科阶段、硕士阶段、职场面试、算法竞赛。