LeetCode 16.最接近的三数之和

发布于:2025-07-30 ⋅ 阅读:(22) ⋅ 点赞:(0)

给你一个长度为 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
解释:与 target 最接近的和是 0(0 + 0 + 0 = 0)。

提示:

3 <= nums.length <= 1000
-1000 <= nums[i] <= 1000
-104^44 <= target <= 104^44

先对nums进行排序,然后固定一个数字,相向双指针找出最接近的三数之和:

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());

        int sum = numeric_limits<int>::max() - 1e4;

        for (int i = 0; i < nums.size() - 2; ++i) {
            int left = i + 1;
            int right = nums.size() - 1;

            // 优化,如果当前固定的数字与其右边最小的两个数字之和curMinSum已经大于target
            // 那么剩余的所有数字中,任意三数之和都大于等于curMinSum
            int curMinSum = nums[i] + nums[left] + nums[left + 1];
            if (curMinSum > target) {
                if (abs(curMinSum - target) < abs(sum - target)) {
                    sum = curMinSum;
                }
                break;
            }

            // 优化,如果当前固定的数字与其右边最大的两个数字之和curMaxSum还是小于target
            // 那么当前固定的数字与其右边所有数字之和都小于curMaxSum
            // 此时只用检查当前固定的数字和其右边最大的两个数字之和即可
            int curMaxSum = nums[i] + nums[right] + nums[right - 1];
            if (curMaxSum < target) {
                if (abs(curMaxSum - target) < abs(sum - target)) {
                    sum = curMaxSum;
                }
                continue;
            }

            while (left < right) {
                int s = nums[left] + nums[right] + nums[i];

                if (abs(s - target) < abs(sum - target)) {
                    sum = s;
                }

                if (s < target) {
                    ++left;
                } else if (s > target) {
                    --right;
                } else {
                    return s;
                }
            }
        }

        return sum;
    }
};

如果nums的长度为n,则此算法时间复杂度为O(n2^22),空间复杂度为O(logn)。


网站公告

今日签到

点亮在社区的每一天
去签到