0%

15.三数之和

排序+双指针

题目要求找出所有和为 $0$ 的不重复三元组。一个很自然的想法是先固定一个数,再去剩余部分中寻找两个数,使三者和为 $0$。

为了高效寻找这两个数,可以先对数组排序。这样在固定 $nums[k]$ 后,就能在区间 $[0, k - 1]$ 中使用双指针查找两数之和是否等于 $-nums[k]$。如果当前和偏小,就让左指针右移;如果当前和偏大,就让右指针左移;如果恰好相等,就记录答案。

排序之后还可以方便地去重。固定的第三个数如果和上一次相同,就直接跳过;找到一组合法答案后,也要分别跳过左侧和右侧的重复元素,避免加入重复三元组。

时间复杂度 $O(n^2)$

空间复杂度 $O(\log n)$

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> ans = new ArrayList<>();
for (int k = nums.length - 1; k >= 2; k--) {
if (k < nums.length - 1 && nums[k] == nums[k + 1]) {
continue;
}
int i = 0, j = k - 1;
while (i < j) {
if (nums[i] + nums[j] == -nums[k]) {
ans.add(Arrays.asList(nums[i], nums[j], nums[k]));
while (i < j && nums[i] == nums[i + 1]) {
i++;
}
while (i < j && nums[j] == nums[j - 1]) {
j--;
}
j--;
i++;
} else if (nums[i] + nums[j] < -nums[k]) {
i++;
} else {
j--;
}
}
}
return ans;
}
}