LeetCode 16.最接近的三数之和(C++)

链接

https://leetcode.cn/problems/3sum-closest/description/

题目

给你一个长度为 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

思路：

先排序，使数据有序，以保证后续可以使用双指针来计算”三数之和“。记录第一次”三数之和“和target的距离(记为sub)，并记录此时”三数之和“的值(记为ret)。当"三数之和“大于target时，比较此时"三数之和“和target的距离和sub的大小关系。当此时"三数之和“和target的距离<sub时，更新sub，并更新ret。然后再缩小”三数之和“的值。当”三数之和“小于target时，比较此时"三数之和“和target的距离和sub的大小关系。当此时"三数之和“和target的距离<sub时，更新sub，并更新ret。然后增加”三数之和“的值。当”三数之和“等于target时，意味着重合，距离为0，因此是最小的，更新ret后返回。

代码实现：

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int ret , sub;
        for(int i = 0 ; i < nums.size(); i++)//遍历取得“三数之和“的第一个数
        {
            int left = i + 1 , right = nums.size() - 1;
            if(i == 0) // 记录首次的 sub 和 ret
            {
                sub = abs(target - (nums[i] + nums[left] + nums[right]));
                ret = (nums[i] + nums[left] + nums[right]);
            }
            while(left < right)
            {
                if(nums[i] + nums[left] + nums[right] < target) 
                {
                    if(abs(target - (nums[i] + nums[left] + nums[right])) < sub) // 更新sub和ret
                    {
                        sub = abs(target - (nums[i] + nums[left] + nums[right]));
                        ret = nums[i] + nums[left] + nums[right];
                    }
                    left++;// 增加 nums[left] 的值，从而使得 "三数之和"的值增大
                }
                else if (nums[i] + nums[left] + nums[right] == target) //更新ret并返回
                {
                    ret = nums[i] + nums[left] + nums[right];
                    return ret;
                }
                else
                {
                    if(abs(target - (nums[i] + nums[left] + nums[right])) < sub) // 更新sub和ret
                    {
                        sub = abs(target - (nums[i] + nums[left] + nums[right]));
                        ret = nums[i] + nums[left] + nums[right];
                    }
                    right--;//减小 nums[right]的值，从而使得“三数之和”的值变小
                }
            }
        }
        return ret;
    }
};