逐行详解 Leetcode 15 --- 三数之和的双指针实现

60 阅读2分钟

Leetcode 15 --- 三数之和

var threeSum = function (nums) {

    const result = []

    nums = nums.sort((x, y) => x - y) // 升序排列

    if (nums[0] > 0) { // 三数之和为 0 ,则第一个数一定小于零
        return result
    }

    for (let i = 0; i < nums.length - 2; i++) {
        // 每次循环中,i 是固定不动的,即三数之和中的第一个数的值是固定的
        
        // 假设第一个值为 value 的元素的索引为 k
        // 则对于固定值 value = nums[k] = nums[k+1] = nums[k+2] = ... = nums[k+n] 来说,
        // nums[k] 已经处理了固定值 value 可以匹配到的所有 第二个数 和 第三个数 的取值组合
        // 即包括了第二个数的值也是 value 的情况
        if (i > 0 && nums[i] === nums[i - 1]) {
            continue
        }
        let L = i + 1 // 左指针 Left
        let R = nums.length - 1 // 右指针 Right
        while (L < R) { // 双指针必须有的判断:L < R (有些场景可能是 L ≤ R)
            const sum = nums[i] + nums[L] + nums[R]
            if (sum === 0) {
                result.push([nums[i], nums[L], nums[R]])
                while (nums[L] === nums[L + 1]) L++ // nums[i] 的值是固定的,要得到新的三数之和的组合,需要更新 L 指针指向的值
                while (nums[R] === nums[R - 1]) R-- // 因为 L 指针指向的值发生了改变,要满足三数之和,R 指针指向的值也应该改变
                // 上一步中, L 指针和 R指针都指向最后一个重复的值
                L++ 
                R-- 
            }
            // 对于 sum !== 0 的情况,注意只能将单个 L 指针或单个 R 指针移动到不重复的下一个值,而不能同时移动
            // 因为满足三数之和为 0 的情况可能是 nums[L] === nums[R],
            // 如果同时移动,则不会发生 nums[L] = nums[R] 的情况,因为最外层的 while 循环要求:L < R
            else if (sum < 0) {
                while (nums[L] === nums[L + 1]) L++  // 与上面的 while 循环目的相同
                L++
            } else {
                while (nums[R] === nums[R - 1]) R--  // 与上面的 while 循环目的相同
                R--
            }
        }
    }

    return result
};