leetcode 三数之和 Java

题干

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

想法

题干要点提取一下就：
1.找到三个元素相加为0
2.不能有重复

最暴力的想法当然是全部枚举出来，但是太蠢了肯定通不过。
于是我们考虑这样查找：
先将nums排序
三个位置
i j k
i从0到nums.length遍历
对于每一个i,如果nums[i]+nums[j]+nums[k]=0,那么就找到一个，加进去完事。
然后j往后,k往前（只有两个数里小的变大，大的变小才有可能=0；

如果nums[i]+nums[j]+nums[k]<0
那么还不够大，j++
反之k–

接下来我们考虑重复的问题
如果nums[j]=nums[j+1]
那么j++往后
同理
如果nums[k]=numsk-1]
那么k–往前
这样就去重

下方部分代码有更好的优化，看代码就行，很简单

java代码：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        if(nums == null){
            return null;
        }
        //如果数组长度小于3，返回一个空集合
        if(nums.length < 3){
            return new ArrayList<>();
        }
        //对数组nums进行排序
        Arrays.sort(nums);
        HashSet<List<Integer>> set = new HashSet<>();
        //让i从数组下标为0开始跑
        for (int i = 0; i < nums.length; i++) {
            //j从i的后一个数开始
            int j = i + 1;
            //k从数组最后一个数（最大的数）往前跑
            int k = nums.length - 1;
            while(j < k){
                if(nums[i] + nums[j] + nums[k] == 0){
                    List<Integer> list = new ArrayList<>();
                    list.add(nums[i]);
                    list.add(nums[j]);
                    list.add(nums[k]);
                    set.add(list);
                    //当有重复的数字出现时，j往前跑，防止结果有重复
                    while(j < k && nums[j] == nums[j + 1]){
                        j++;
                    }
                    while(j < k && nums[k] == nums[k - 1]){
                        k--;
                    }
                    j++;
                    k--;
                }else if(nums[i] + nums[j] + nums[k] < 0){
                    j++;
                }else{
                    k--;
                }
            }
        }
        return new ArrayList<>(set);
    }
}

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
       int n = nums.length;
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        for (int i = 0; i < n; i++) {
            int k = i + 1;
            int j = n - 1;

            int value = nums[i];
            if (value > 0) {
                break;
            }

            while (k < j) {
                if (nums[k] + nums[j] + value == 0) {
                    List<Integer> list = new ArrayList<>();
                    list.add(value);
                    list.add(nums[k]);
                    list.add(nums[j]);
                    result.add(list);
                    while (k + 1 < j && nums[k] == nums[k + 1]) {
                        k++;
                    }
                    while (j - 1 > k && nums[j] == nums[j - 1]) {
                        j--;
                    }
                    k++;
                    j--;
                } else if (nums[k] + nums[j] + value < 0) {
                    k++;
                } else {
                    j--;
                }
            }

            while (i < n-1 && nums[i] == nums[i + 1]) {
                i++;
            }
        }
        return result;
        
    }
}

class Solution {
     public static List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> ans = new ArrayList();
        int len = nums.length;
        if(nums == null || len < 3) return ans;//nums为空返回空 不足三个数就返回空
        Arrays.sort(nums); // 排序
        for (int i = 0; i < len ; i++) {//i是最左边的最小的数
            if(nums[i] > 0) break; // 如果当前数字大于0，则三数之和一定大于0，所以结束循环
            if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
            int L = i+1;
            int R = len-1;
            while(L < R){
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0){
                    ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
                    while (L<R && nums[L] == nums[L+1]) L++; // 去重
                    while (L<R && nums[R] == nums[R-1]) R--; // 去重
                    L++;
                    R--;
                }
                else if (sum < 0) L++;
                else if (sum > 0) R--;
            }
        }        
        return ans;
    }
}

思路代码参考https://blog.csdn.net/qq_39241239/article/details/82387016