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

哈喽各位,我是前端小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,算两数之和。

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

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

相关推荐
黎阳之光19 分钟前
黎阳之光:视频孪生领跑者,铸就中国数字科技全球竞争力
大数据·人工智能·算法·安全·数字孪生
skywalker_1125 分钟前
力扣hot100-3(最长连续序列),4(移动零)
数据结构·算法·leetcode
6Hzlia26 分钟前
【Hot 100 刷题计划】 LeetCode 17. 电话号码的字母组合 | C++ 回溯算法经典模板
c++·算法·leetcode
wfbcg1 小时前
每日算法练习:LeetCode 209. 长度最小的子数组 ✅
算法·leetcode·职场和发展
哟哟耶耶1 小时前
vue3-单文件组件css功能(:deep,:slotted,:global,useCssModule,v-bind)
前端·javascript·css
是罐装可乐1 小时前
深入理解“句柄(Handle)“:从浏览器安全到文件系统访问
前端·javascript·安全
_日拱一卒1 小时前
LeetCode:除了自身以外数组的乘积
数据结构·算法·leetcode
计算机安禾1 小时前
【数据结构与算法】第36篇:排序大总结:稳定性、时间复杂度与适用场景
c语言·数据结构·c++·算法·链表·线性回归·visual studio
SatVision炼金士1 小时前
合成孔径雷达干涉测量(InSAR)沉降监测算法体系
算法
wuweijianlove1 小时前
算法稳定性与数值误差传播研究的技术2
算法