LeetCode15-三数之和

一、题目

给你一个包含 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;
    }
}