【LeetCode】刷题总结 - 15. 三数之和

yubinCloud

已于 2024-02-04 21:48:24 修改

阅读量633

点赞数 3

分类专栏： LeetCode 文章标签： leetcode 算法

于 2024-02-04 21:34:37 首次发布

本文链接：https://blog.csdn.net/qq_45668004/article/details/136032516

版权

LeetCode 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

本文介绍了如何使用排序和双指针方法解决LeetCode中的三数之和问题，通过不断调整左右指针和中间指针找到所有满足条件的不重复整数组合。

摘要由CSDN通过智能技术生成

15. 三数之和 | LeetCode

思路

首先对 nums 进行排序，然后设置三个指针：left、mid、right：

left 从最左边开始，依次向后遍历
每次固定住 left 后，就化为一个 2sum 问题：
- mid = left + 1，right = nums.length - 1，计算 sum
- 如果 sum 大于 0，则 right 向后走；如果 sum 小于 0，则 mid 向前走；等于 0 则加入结果集并让 right 和 mid 均走一步

注意点：

⚠️ left、mid、right 在向前或者向后走一步时，都要直接跳过重复值，防止产生重复解
在 nums[left] 的值大于 0 的情况下，三数之和不可能再为 0，所以可以提前结束

参考思路：

一个函数秒杀 2Sum 3Sum 4Sum 问题 | labuladong
三数之和（排序+双指针，易懂图解）| LeetCode

代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        int left = 0, right = nums.length - 1, mid = 0;
        List<List<Integer>> results = new ArrayList<>();
        
        while (true) {
            // special case
            if (left > nums.length - 2) {
                return results;
            }
            if (left != 0 && nums[left] == nums[left - 1]) {
                left++;
                continue;
            }
            if (nums[left] > 0) {  // left 的值大于 0，表示 3sum 不可能再是 0
                return results;
            }
            // 固定住 left，化为解 2sum 问题
            mid = left + 1;
            right = nums.length - 1;
            while (mid < right) {
                int sum = nums[left] + nums[mid] + nums[right];
                if (sum > 0) {
                    right--;
                    while (right > mid && nums[right] == nums[right + 1]) {  // 跳过重复值
                        right--;
                    }
                } else if (sum < 0) {
                    mid++;
                    while (mid < right && nums[mid] == nums[mid - 1]) {  // 跳过重复值
                        mid++;
                    }
                } else {
                    results.add(List.of(nums[left], nums[mid], nums[right]));
                    right--;
                    while (right > mid && nums[right] == nums[right + 1]) {  // 跳过重复值
                        right--;
                    }
                    mid++;
                    while (mid < right && nums[mid] == nums[mid - 1]) {  // 跳过重复值
                        mid++;
                    }
                }
            }
            left++;
        }
    }
}