算法:三数之和(js)

题目: 力扣

思路:排序+三指针,从小到大排序后,当前指针从第一个元素开始,在当前指针的右边进行首尾指针移动,cur指针指向的值+left指向的值+right指向的值 === 0时,res中存入该组值。

代码:

var threeSum = function(nums) {
    const res = []
    const arr = nums.sort((a, b) => a - b)
    for (let i = 0; i < arr.length - 2; i ++) {
        let left = i + 1, right = arr.length - 1, cur = arr[i]
        if (cur > 0) break
        if (i - 1 >= 0 && cur === arr[i - 1]) continue // 当前与前一个一样就不用继续走了
        while (left < right) {
            const leftVal = arr[left], rightVal = arr[right]
            if (cur + leftVal + rightVal === 0) {
            res.push([cur, leftVal, rightVal])
            // 此时要把left和right移到下一个不一样的数
            while(left < right && arr[left] === leftVal) left ++
            while(left < right && arr[right] === rightVal) right --
          } else if (cur + arr[left] + arr[right] < 0) {
             left ++
          } else {
             right --
          }
        }
    }
    return res
};

猜你喜欢

转载自blog.csdn.net/qq_43119912/article/details/124368830