[数据结构与算法]leetcode 三数之和

题目描述

给你一个包含 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 = []
输出：[]
输入：nums = [0]
输出：[]

思路

源于三数之和，我的理解注释都已经写到代码中

代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res=new ArrayList<>();
        //最后两个开始的话，也就拼不成3个数了
        for(int i=0;i<nums.length-2;i++){
            int l=i+1;
            int r=nums.length-1;
            //nums[r]>nums[l]>nums[i] 如果nums[i]>0 直接跳出,因为num是排完序的，后面也不需要考虑了  
            if(nums[i]>0){break;}
            //直接将循环的值跳过
            if(i>0 && nums[i]==nums[i-1]){continue;}
            while(l<r){
                int sum=nums[i]+nums[l]+nums[r];
                if(sum<0){
                //相当于l移动到下一个不同的值
                while(l<r && nums[l]==nums[++l]);
                }else if(sum>0){
                //相当于r移动到下一个不同的值
                while(l<r && nums[r]==nums[--r]);
                }else{
                    res.add(Arrays.asList(nums[i],nums[l],nums[r]));
                    //相当于l移动到下一个不同的值
                    //相当于r移动到下一个不同的值
                    while(l<r && nums[r]==nums[--r]);
                    while(l<r && nums[l]==nums[++l]);
                }
            }
        }
        return res;
    }
}

另一种方式 暴力求和（简单易想，但是复杂度太大）

思路

三层for循环求出所有可能，但是可能会有很多重复集合，
需要使用hashset来进行去重