maninbule/DailyQuestion

【每日一题】 leetcode16 最接近的三数之和

Opened this issue · 1 comments

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

示例

给定数组 nums = [-1,2,1,-4], 和 target = 1.

与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

首先直接暴力O(n^3)是一定会超时的,所以我们的目的是把此问题转换成O(n^2)
比较A+B+C于target的关系,如果A+B+C <target,可以让A+B+C变大一些,就更加接近target
A+B+C >target,就让A+B+C变小一些,就更加接近target
但是现在有A,B,C三个变量,没法一步到位,所以我们可以想到先循环一遍nums,固定一个变量,
然后只有B和C了,我们可以对nums从小到大排序,B是从开始往后移动逐渐变大,C是从后往前移动逐渐变小,所以此时A+B+C <target,就只需要将B往后移动就增大了,A+B+C >target,就只需要将C往前移动就减小了。

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        int len = nums.size();
        sort(nums.begin(),nums.end());
        int res = 2e9;
        for(int i = 0;i<len;i++){
            int l = i+1,r = len-1;//左端从i+1就行了,可以防止重复计算
            while(l<r){
                int cur = nums[i]+nums[l]+nums[r];
                if(cur<target) l++;
                else if(cur>target) r--;
                else return cur;
                if((int)abs(cur-target) < (int)abs(res-target)){
                    res = cur;
                }
            }
        }
        return res;
    }
};