双指针专题(三):去重的艺术——「三数之和」

哈喽各位,我是前端小L。

场景想象:

给你一个数组 [-1, 0, 1, 2, -1, -4]。

我们要找出所有和为 0 的三个数 [a, b, c]。

  • 我们可以找到 [-1, 0, 1]

  • 还可以找到 [-1, 2, -1](排序后是 [-1, -1, 2])。

  • 难点 :数组里有两个 -1,如果我们不加控制,代码很可能会把第一个 -1 开头的 [-1, 0, 1] 算一次,再把第二个 -1 开头的 [-1, 0, 1] 又算一次。这就是重复

力扣 15. 三数之和

https://leetcode.cn/problems/3sum/

题目分析:

  • 输入 :整数数组 nums

  • 输出 :所有和为 0 的不重复三元组。

  • 复杂度要求 :暴力法是 O(N\^3),肯定超时。我们需要优化到 O(N\^2)

核心思维:排序 + 固定一个,找另外两个

既然双指针擅长解决"两数之和",那我们能不能把"三数之和"降维打击?

策略:

  1. 先排序 :这是去重和使用双指针的前提!(例如 [-4, -1, -1, 0, 1, 2])。

  2. 遍历固定位 (i) :我们遍历数组,固定第一个数 nums[i]

  3. 双指针找两数 :剩下的问题就变成了------"在 i 后面的数组中,找到两个数 LeftRight,使得 nums[Left] + nums[Right] = -nums[i]"。

去重逻辑(最关键):

  1. 外层去重(针对 i):

    如果 nums[i] === nums[i-1],说明这个数刚才已经作为"第一个数"处理过了,再处理一遍肯定会得到一样的结果。跳过!

  2. 内层去重(针对 Left 和 Right):

    当我们找到了一个合法的组合后,如果 nums[Left] === nums[Left+1],说明下一个数还是一样,会导致结果重复。跳过! 右边同理。

算法流程 (JavaScript)

  1. 排序nums.sort((a, b) => a - b)

  2. 外层循环i0n-2

    • 如果 nums[i] > 0:因为已经排序,第一个数大于0,后面不可能凑出 0 了,直接 break

    • 去重if (i > 0 && nums[i] === nums[i - 1]) continue;

    • 双指针启动L = i + 1, R = n - 1

      • 计算 sum = nums[i] + nums[L] + nums[R]

      • sum > 0:数太大了,R 往左移。

      • sum < 0:数太小了,L 往右移。

      • sum === 0

        • 记录答案res.push([nums[i], nums[L], nums[R]])

        • 内层去重while (L < R && nums[L] === nums[L+1]) L++; (跳过重复的L)

        • 内层去重while (L < R && nums[R] === nums[R-1]) R--; (跳过重复的R)

        • 双双移动L++, R-- (寻找下一组)。

代码实现

JavaScript

复制代码
/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    const result = [];
    // 1. 必须排序
    nums.sort((a, b) => a - b);
    const len = nums.length;

    for (let i = 0; i < len; i++) {
        // 剪枝:如果第一个数已经大于0,后面都是正数,不可能凑成0
        if (nums[i] > 0) break;

        // 2. 外层去重:如果我们遇到了和上一次一样的数字,跳过
        // 注意是 i > 0,防止 i-1 越界
        if (i > 0 && nums[i] === nums[i - 1]) continue;

        let L = i + 1;
        let R = len - 1;

        while (L < R) {
            const sum = nums[i] + nums[L] + nums[R];

            if (sum === 0) {
                // 找到一组,加入结果
                result.push([nums[i], nums[L], nums[R]]);
                
                // 3. 内层去重:找到答案后,看看左右指针旁边是不是有重复的
                // 如果有,一直跳过,直到指向最后一个重复元素的下一位
                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 { // sum > 0
                // 和太大,右指针左移变小
                R--;
            }
        }
    }
    return result;
};

深度辨析:为什么去重是 nums[i] === nums[i-1]

很多同学会纠结:是判断 nums[i] == nums[i-1] 还是 nums[i] == nums[i+1]

  • 情况 A (nums[i] == nums[i-1]):

    比如 [-1, -1, 2]。

    • i=0 (第一个 -1):处理一遍,找到了 [-1, -1, 2]

    • i=1 (第二个 -1):我们回头看,发现前一个也是 -1。如果我们再处理,就会重复寻找。所以跳过。

    • 这是正确的逻辑:只要我前面那个"长得跟我一样"的兄弟已经处理过了,我就不用再处理了。

  • 情况 B (nums[i] == nums[i+1]):

    如果我们判断的是后一个。

    • i=0 (第一个 -1):发现后面也是 -1,直接跳过?

    • 错! 这样你就漏掉了 [-1, -1, 2] 这一组情况(这组情况需要用到两个 -1)。

    • 结论 :我们去重的是"作为第一个加数的重复",而不是禁止三元组内部有重复数字。

总结

这道题是双指针的里程碑。

掌握了它,你就掌握了 "排序 + 降维 + 双指针夹逼 + 细致去重" 的完整套路。


下一题预告:四数之和

如果面试官觉得三数之和太简单,可能会随手丢给你一道 LC 18. 四数之和

  • 题目:找到 a + b + c + d = target

  • 思路:套娃

    • 三数之和是:固定 i,算两数之和。

    • 四数之和是:固定 k,再固定 i,算两数之和。

  • 难点 :多了一层循环,也就多了一层剪枝去重的逻辑。

准备好再加一层循环了吗?下期见!

相关推荐
web小白成长日记2 小时前
在Vue样式中使用JavaScript 变量(CSS 变量注入)
前端·javascript·css·vue.js
C_心欲无痕2 小时前
react - useImperativeHandle让子组件“暴露方法”给父组件调用
前端·javascript·react.js
智者知已应修善业3 小时前
【求等差数列个数/无序获取最大最小次大次小】2024-3-8
c语言·c++·经验分享·笔记·算法
霖鸣3 小时前
Minecraft通过kubejs进行简单魔改
javascript
LYFlied3 小时前
【每日算法】LeetCode 416. 分割等和子集(动态规划)
数据结构·算法·leetcode·职场和发展·动态规划
JackieDYH3 小时前
HTML+CSS+JavaScript实现图像对比滑块demo
javascript·css·html
多米Domi0113 小时前
0x3f 第19天 javase黑马81-87 ,三更1-23 hot100子串
python·算法·leetcode·散列表
历程里程碑4 小时前
滑动窗口最大值:单调队列高效解法
数据结构·算法·leetcode
課代表4 小时前
从初等数学到高等数学
算法·微积分·函数·极限·导数·积分·方程