题目描述
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
示例
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路
- 暴力法搜索为 O(N3)时间复杂度,可通过双指针动态消去无效解来优化效率。
- 双指针法铺垫: 先将给定 nums 排序,复杂度为 O(NlogN)。
- 双指针法思路: 固定 33 个指针中最左(最小)数字的指针 k,双指针 i,j 分设在数组索引
(k+1, len(nums)-1)
两端,通过双指针交替向中间移动,记录对于每个固定指针 k 的所有满足nums[k] + nums[i] + nums[j] == 0
的 i,j 组合:- 当 n
ums[k] > 0
时直接break跳出:因为nums[j] >= nums[i] >= nums[k] > 0
,即 3个数字都大于 0 ,在此固定指针 k 之后不可能再找到结果了。 - 当
k > 0
且nums[k] == nums[k - 1]
时即跳过此元素nums[k]:因为已经将nums[k - 1]
的所有组合加入到结果中,本次双指针搜索只会得到重复组合。 - 当
i < j
时循环计算s = nums[k] + nums[i] + nums[j]
,并按照以下规则执行双指针移动:- 当
s < 0
时,i += 1
并跳过所有重复的nums[i]
; - 当
s > 0
时,j -= 1
并跳过所有重复的nums[j]
; - 当
s == 0
时,记录组合[k, i, j]
至res,执行i += 1
和j -= 1
并跳过所有重复的nums[i]
和nums[j]
,防止记录到重复组合。
- 当
- 当 n
- 复杂度分析:
- 时间复杂度 O(N^2):其中固定指针k循环复杂度O(N),双指针 i,j 复杂度 O(N)。
- 空间复杂度 O(1):指针使用常数大小的额外空间。
代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> res = new ArrayList<>();
for (int k = 0; k < nums.length - 2; k++) {
if (nums[k] > 0) break;
if (k > 0 && nums[k] == nums[k-1]) continue;
int i = k + 1, j = nums.length - 1;
while (i < j) {
int sum = nums[k] + nums[i] + nums[j];
if (sum < 0) {
i++;
} else if (sum > 0) {
j--;
} else {
res.add(Arrays.asList(nums[k], nums[i], nums[j]));
while(i < j && nums[i] == nums[++i]);
while(i < j && nums[j] == nums[--j]);
}
}
}
return res;
}
}