【重点】【双指针】15. 三数之和

Allenlzcoder

已于 2024-01-23 20:11:48 修改

阅读量418

点赞数 9

分类专栏：力扣Top100 labuladong的算法小抄笔记文章标签：三数之和双指针

于 2023-12-03 12:45:58 首次发布

本文链接：https://blog.csdn.net/Allenlzcoder/article/details/134763481

版权

力扣Top100 同时被 2 个专栏收录

100 篇文章 0 订阅

订阅专栏

labuladong的算法小抄笔记

77 篇文章 0 订阅

订阅专栏

题目
注意：三数之和属于双指针问题，二数之和属于哈希问题。
版本1代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        int n = nums.length;
        if (n < 3) {
            return res;
        }
        Arrays.sort(nums);
        for (int i = 0; i < n - 2; ++i) {
            if (nums[i] > 0) {
                break;
            }
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int j = i + 1, k = n - 1;
            while (j < k) {
                int tmp = nums[i] + nums[j] + nums[k];
                if (tmp == 0) {
                    res.add(new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k])));
                    while (j + 1 < k && nums[j + 1] == nums[j]) {
                        ++j;
                    }
                    while (k - 1 > j && nums[k - 1] == nums[k]) {
                        --k;
                    }
                    ++j;
                    --k;
                } else if (tmp < 0) {
                    ++j;
                } else {
                    --k;
                }
            }
        }

        return res;
    }
}

版本2代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        if (nums.length < 3) {
            return res;
        }
        Arrays.sort(nums);
        int i = 0, j, k;
        while (i < nums.length - 2) {
            j = i + 1;
            k = nums.length - 1;
            while (j < k) {
                if (nums[i] + nums[j] + nums[k] == 0) {
                    List<Integer> tmp = new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k]));
                    res.add(tmp);
                    ++j;
                    --k;
                    while (j < k && nums[j] == nums[j - 1]) {
                        ++j;
                    }
                    while (j < k && nums[k] == nums[k + 1]) {
                        --k;
                    }
                } else if (nums[i] + nums[j] + nums[k] > 0) {
                    --k;
                } else {
                    ++j;
                }
            } 
            ++i;
            while (i < nums.length && nums[i] == nums[i - 1]) {
                ++i;
            }
        }

        return res;
    }
}