给你一个包含 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] 输出:[]
两层for循环就可以确定 a 和b 的数值了,可以使用哈希法来确定 0-(a+b) 是否在 数组里出现过,其实这个思路是正确的,但是我们有一个非常棘手的问题,就是题目中说的不可以包含重复的三元组。
把符合条件的三元组放进vector中,然后再去重,这样是非常费时的,很容易超时,也是这道题目通过率如此之低的根源所在。
去重的过程不好处理,有很多小细节,如果在面试中很难想到位。
时间复杂度可以做到$O(n^2)$,但还是比较费时的,因为不好做剪枝操作。
大家可以尝试使用哈希法写一写,就知道其困难的程度了。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list = new ArrayList<>();
int n = nums.length;
Arrays.sort(nums);
for (int i =0; i < n ; i++) {
if (nums[i]>0){
return list;
}
//去掉重复的结果
if (i>0 && nums[i] == nums[i-1]){
continue;
}
int left = i+1;
int right = n-1;
//对数组排序后使用双指针
while (left < right) {
//如果三数之和大于0说明nums[right]大了,应该right--
if (nums[i] + nums[left] + nums[right] > 0) {
right--;
//如果三数之和小于0说明nums[left]小了,所以left++;
}else if (nums[i] + nums[left] + nums[right] < 0) {
left++;
}else {//三数之和等于0就添加到集合中去
list.add(Arrays.asList(nums[i],nums[left],nums[right]));
//还需要考虑去重
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
right--;
left++;
}
}
}
return list;
}
}