一、题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
二、自己的思路
最先想到的就是dfs,就像组合总和这题一样,找到一个组合添加到List里,回溯一下就好了嘛
但最终会遇到需要去重的问题。以下是自己的做法:
class Solution {
List<List<Integer>> res;
public List<List<Integer>> threeSum(int[] nums) {
res = new ArrayList<>();
dfs(nums,0,0,3,new ArrayList<Integer>());
//拿到结果后,res里是有重复部分的,怎么去重就因人而异了
...去重...
return res;
}
public void dfs(int[] nums,int index,int sum,int count,List<Integer> list){
if(sum == 0 && count == 0){
res.add(new ArrayList<>(list));
return;
}
for(int i = index; i < nums.length; i++){
if(count == 0) return;
list.add(nums[i]);
//i+1是关键,一开始踩了个坑
dfs(nums,i+1,sum+nums[i],count-1,list);
list.remove(list.size()-1);
}
}
}
三、最常用的解法
先把原数组排序,首先固定一个数,在剩下的数中找到两个数,使得三数之和为0,就添加到list里面。把问题转化为两数之和问题。这就避免了拿到结果后还需要去重的问题,还是很牛逼的。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums);
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){
res.add(new ArrayList<>(Arrays.asList(nums[k],nums[i],nums[j])));
while(i < j && nums[i] == nums[++i]);
while(i < j && nums[j] == nums[--j]);
}else if(sum < 0){
while(i < j && nums[i] == nums[++i]);
}else{
while(i < j && nums[j] == nums[--j]);
}
}
}
return res;
}
}