【中等】给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
/**
* 三数之和,如果不使用特殊点的处理方式,逃脱不了三重遍历,那时间复杂度必然是O(n^3)
* 参考两数之和,先将数组排序变成升序的
* 想尽办法转化为两数之和,两数之和的复杂度是n, 那再来一个数,顶多是n^2
* 先固定一个指针i, 从0开始。那就是在右侧再找两个数使得low+high = 0 - nums[i]不就可以了吗。
* 不就转换成在一个数组里,还是升序的数组,找两个数,使得两数之和等于目标数,那就可以使用典型的双指针法,复杂度是n
*
* 当然这里面,i, low,high指针对应的数不能重复了
*/
public List<List<Integer>> method2(int[] nums) {
Arrays.sort(nums);
int length = nums.length;
List<List<Integer>> res = new ArrayList<>();
for (int i = 0; i < length - 2; i++) {
if (nums[i] > 0) return res;
if (i > 0 && nums[i] == nums[i-1]) continue;
int low = i + 1, high = length - 1, sum = -nums[i];
// 下面就是找两个数的和等于sum
while (low < high) {
if (nums[low] + nums[high] == sum) {
res.add(Arrays.asList(nums[i], nums[low], nums[high]));
// 将low,high向中间逼近,并且避免重复
while (low < high && nums[low] == nums[low+1]) low++;
while (low < high && nums[high] == nums[high-1]) high--;
low++;
high--;
}
else if (nums[low] + nums[high] < sum) {
while (low < high && nums[low] == nums[low+1]) low++;
low++;
}
else {
while (low < high && nums[high] == nums[high-1]) high--;
high--;
}
}
}
return res;
}