【LeetCode】15. 三数之和_双指针

题目描述

15. 三数之和

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

2022年12月26日08:54:32

class Solution {

    public List<List<Integer>> threeSum(int[] nums) {
        int n = nums.length;
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<>();
        // 枚举 a
        for (int first = 0; first < n; ++first) {
            // 如果a 大于0的时候，则不可能满足三数相加等于0
            if (nums[first] > 0){
                break;
            }
            // 需要和上一次枚举的数不相同
            if (first > 0 && nums[first] == nums[first - 1]) {
                continue;
            }
            // c 对应的指针初始指向数组的最右端
            int third = n - 1;
            int target = -nums[first];
            // 枚举 b
            for (int second = first + 1; second < n; ++second) {
                // 需要和上一次枚举的数不相同
                if (second > first + 1 && nums[second] == nums[second - 1]) {
                    continue;
                }
                // 需要保证 b 的指针在 c 的指针的左侧  && 存在收缩条件：b和c相加是大于目标值
                while (second < third && nums[second] + nums[third] > target) {
                    // 移动右指针
                    --third;
                }
                // 如果指针重合，随着 b 后续的增加
                // 就不会有满足 a+b+c=0 并且 b<c 的 c 了，可以退出循环
                if (second == third) {
                    break;
                }
                if (nums[second] + nums[third] == target) {
                    List<Integer> list = new ArrayList<>();
                    list.add(nums[first]);
                    list.add(nums[second]);
                    list.add(nums[third]);
                    ans.add(list);
                }
            }
        }
        return ans;
    }

2023-8-28 10:06:42

第二次写，还是不会。Orz

    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        int length = nums.length;
        // 因为是三数之和，所以范围就是 [0,n-2)
        for(int k = 0; k < length - 2; k++){
            // 因为要找 `nums[i] + nums[j] + nums[k] == 0`
            // 由于数组已经排好序了，所以当 这个数 大于0的时候，就不可能会满足后序的数 与 这个数 相加等于0的情况
            if(nums[k] > 0) {
                break;
            }
            // `返回所有和为 0 且不重复的三元组。`
            // 所谓的重复情况就是：如 [-1 -1 0 1 ] ,那么 [-1(index = 0) , 0 ,1] 和 [-1(index = 1) , 0 ,1]
            // 所以 为了 保证能够找到不重复的三元组，跳过重复数字
            if(k > 0 && nums[k] == nums[k - 1]) {
                continue;
            }
            // 将三数之和 转化为 两数之和
            int i = k + 1, j = length - 1;
            while(i < j){
                int sum = nums[k] + nums[i] + nums[j];
                // 以 nums[k] 为固定值，转化为两数之和
                // 如果相加的数 小于0，也就是等于最小的数太小了，需要下标右移，右移之前，跳过重复数字
                if(sum < 0){
                    while(i < j && nums[i] == nums[++i]);
                }
                // 如果相加的数 大于0，也就是等于最大的数太大了，需要下标左移，左移之前，跳过重复数字
                else if (sum > 0) {
                    while(i < j && nums[j] == nums[--j]);
                }
                // 如果相加的数 等于0
                else {
                    // 找到一组答案
                    res.add(new ArrayList<Integer>(Arrays.asList(nums[k], nums[i], nums[j])));
                    // 下标右移,右移之前，跳过重复数字
                    while(i < j && nums[i] == nums[++i]);
                    // 下标左移，左移之前，跳过重复数字
                    while(i < j && nums[j] == nums[--j]);
                }
            }
        }
        return res;
    }