[Leecode]三数之和15. 3Sum Given an array nums of n integers, are there elements a, b, c in nums

原题

链接：三数之和

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

题目大意
给定一个整数型数组，找出符合a + b + c = 0的情况。

思路1:

我首先想到的是深度搜索DFS
给定数组作为一个可选择的路径，然后再创建有三个元素的数组存放已选择的数字。
以下贴出代码

class Solution {
    
    public List<List<Integer>> threeSum(int[] nums) {
        Set<List<Integer>> result =new HashSet<>();
        if(nums.length==0)return new ArrayList<>();
        Arrays.sort(nums);
        dfs(nums,new int[3],0,0,result);
        return new ArrayList(result);
    }

    private static void dfs(int[] nums,int[] visited,int index,int deep,Set<List<Integer>> result){
        if(deep==3&&visited[0]+visited[1]+visited[2]==0){
            ArrayList<Integer> success = new ArrayList<>();
            success.add(visited[0]);
            success.add(visited[1]);
            success.add(visited[2]);
            result.add(success);
        }
        if(deep>=3)return;

        for(int i=index;i<nums.length;i++){
            visited[deep++]=nums[i];
            dfs(nums,visited,++index,deep,result);
            visited[--deep]=0;
        }
    }
}

代码讲解
首先判空
如果是空数组直接返回空.
然后写出dfs函数
第一步判断深度是否为三,然后三个数字加起来是否符合题意等于0,
符合则加入到result中.
然后就是我们dfs的主要思想

for(选择列表){
	做出选择
	dfs()
	撤销选择
}

不过由于深搜比较耗时,提交后提示超出时间限制。

思路2：

双指针解法
先排序 然后固定一个数，剩下两个数使用左右指针指定

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);
        int num=nums.length;
        for(int i=0;i<num;i++){
            if(nums[i]>0)break;
            if(i > 0 && nums[i] == nums[i-1]) continue; 
            resolve(nums,i,result);
            
        }
        return result;
    }

    private static void resolve(int[] nums,int index,List<List<Integer>> result){
        int left=index+1;
        int right=nums.length-1;

        while(left<right){
            int sum=nums[index]+nums[left]+nums[right];
            if(sum==0){
                result.add(Arrays.asList(nums[index],nums[left],nums[right]));
                while (left<right && nums[left] == nums[left+1]) left++; // 去重
                while (left<right && nums[right] == nums[right-1]) right--; // 去重
                left++;
                right--;
            }else if(sum<0){
                left++;
            }else if(sum>0){
                right--;
            }

        }
    }
}

代码讲解
我们每次固定一个数
这个数下标设为index
左指针left下标设定为index+1
右指针right下标设为数组最后一个数

当三数相加时小于零，则说明需要增大其中一个数，index是固定的，右指针是最大元素（因为我们已经将数组排序过了），所以将左指针往右移动一个位置，left++

当三数相加时大于零，则说明需要减小其中一个数，index是固定的，left是最小的，所以将右指针往左移动一个位置，right–

三数相加时等于零，说明符合要求，由于数组中可能存在重复的元素，所以我们需要先判断left是否与left+1重复，right是否和right-1重复，如果重复则直接跳过遍历这些重复元素，防止最后结果集中有重复的答案。