16. 3Sum Closest

这道题又是上一题3Sum的变形,3Sum是求出所有三个数相加和为0的情况,这个题是求出是三个数的和最接近target的情况,返回这个和。

我的思路还是用了3Sum的方法,遍历数组加2Sum二分查找。但是有些地方没注意,也调了好久。一个是如果2Sum直接等于target-nums[i],说明找到了三个数的和正好等于target,这个时候函数可以直接返回target,不用继续了,因为不可能有更好的方案了。再一个就是,我设了一个diff用来看什么时候更新res,这个diff必须用绝对值来更新,也就是abs那里,一开始忘了加。

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        int diff = INT_MAX;
        int res = 0;
        for(int i = 0; i < nums.size() - 2; ++i){
            if(i > 0 && nums[i] == nums[i-1])    continue;
            int left = i+1, right = nums.size()-1;
            while(left < right){
                if(diff > abs(nums[left] + nums[right] + nums[i] - target)){
                        diff = abs(nums[left] + nums[right] + nums[i] - target);
                        res = nums[left] + nums[right] + nums[i];
                }
                if(nums[left] + nums[right] < target - nums[i]){
                    while(left+1 < nums.size() && nums[left+1] == nums[left])   left++;
                    left++;
                }else if(nums[left] + nums[right] > target - nums[i]){
                    while(right-1 > 0 && nums[right-1] == nums[right]) right--;
                    right--;
                }else{
                    return target;
                }
            }
        }
        return res;
    }
};

这个题也可以不用去重复的代码,看别人写的就很简单:

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        int closest = nums[0] + nums[1] + nums[2];
        int diff = abs(closest - target);
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size() - 2; ++i) {
            int left = i + 1, right = nums.size() - 1;
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];
                int newDiff = abs(sum - target);
                if (diff > newDiff) {
                    diff = newDiff;
                    closest = sum;
                }
                if (sum < target) ++left;
                else --right;
            }
        }
        return closest;
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值