leetCode算法—16. 最接近的三数之和

15. 三数之和

给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数,使它们的和与 target 最接近。

返回这三个数的和。

假定每组输入只存在恰好一个解。

  • 示例 1:
    输入:nums = [-1,2,1,-4], target = 1
  • 输出:2

解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

  • 示例 2:
    输入:nums = [0,0,0], target = 1
    输出:0

  • 提示:
    3 <= nums.length <= 1000
    -1000 <= nums[i] <= 1000
    -104 <= target <= 104

解法:

const threeSumClosest = function (nums, target) {
    nums.sort((a, b) => a - b);
    const n = nums.length;
    let ans = 0;
    let minDiff = Number.MAX_SAFE_INTEGER;
    for (let i = 0; i < n - 2; i++) {
        const x = nums[i];
        if (i > 0 && x === nums[i - 1]) continue;
        // 优化一
        let s = x + nums[i + 1] + nums[i + 2];
        if (s > target) { // 后面无论怎么选,选出的三个数的和不会比 s 还小
            if (s - target < minDiff) ans = s;
            break;
        }

        // 优化二
        s = x + nums[n - 2] + nums[n - 1];
        if (s < target) { // x 加上后面任意两个数都不超过 s,所以下面的双指针就不需要跑了
            if (target - s < minDiff) {
                minDiff = target - s;
                ans = s;
            }
            continue;
        }

        // 双指针
        let left = i + 1, right = n - 1;
        while (left < right) {
            s = x + nums[left] + nums[right];
            if (s === target) return target;
            if (s > target) {
                right--;
            } else {
                left++;
            }

            if (s - target < minDiff) { // s 与 target 更近
                minDiff = Math.abs(s - target);
                ans = s;
            }
        }
    }
    return ans;
};

上一篇:leetCode算法—15. 三数之和
下一篇:

你可能感兴趣的:(前端算法,算法,leetcode,数据结构)