15.三数之和

15.三数之和

题目分析

• 去重：由于是经过排序的,那么相等的一定是挨在一起的,2位置的-1,他只向他后面找能组合成功的数，但是1位置的-1已经找过了,并且1位置,他找的范围还要更大,也就是2位置时其实是可以跳过的
  
• l,r 如果扫描到和前一次的元素相等，即重复情况就要跳过

Solution

    public List<List<Integer>> threeSum(int[] nums) {
        if (nums == null) return null;
        List<List<Integer>> res = new ArrayList<>();
        if (nums.length < 3) return res;

        // 排序
        Arrays.sort(nums);

        // i用来扫描三元组的第一个元素
        int lastIdx = nums.length - 3;
        int lastR = nums.length - 1;
        for (int i = 0; i <= lastIdx; i++) {
            // 当前元素和前一个相同，重复情况
            if (i > 0 && nums[i] == nums[i - 1]) continue;
            // l为小的那个，r为大的那个
            int l = i + 1, r = lastR, remain = -nums[i];
            while (l < r) { // 退出循环，i和其后面的数字之和都计算过了
                int sumLr = nums[l] + nums[r];
                if (sumLr == remain) { // 找到了符合条件的三元组
                    res.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 (sumLr < remain) { // 三数之和偏小
                    l++;
                } else { // 三数之和偏大
                    r--;
                }
            }
        }
        return res;
    }

Reference：小码哥MJ