leetcode 16. 最接近的三数之和

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

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

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

本题与上一题leetcode 15.三数之和https://blog.csdn.net/u011750466/article/details/80218002解法思路相同,程序稍有改动。

1.去重版

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int sum=0,close=INT_MAX,m=0,result=0;//转换为两数之和的问题
        for(int i=0;i<nums.size()-2;++i){
            m=target-nums[i];
            int l=i+1,r=nums.size()-1;//双指针法
            while(l<r){
                sum=nums[l]+nums[r];
                if(sum<m){
                    if(m-sum<close){
                        close=m-sum;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    ++l;
                    while(l<r&&nums[l]==nums[l-1]) ++l;//去重
                }
                else if(sum>m){
                    if(sum-m<close){
                        close=sum-m;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    --r;
                    while(l<r&&nums[r]==nums[r+1]) --r;//去重
                }
                else return target;//如果相等直接返回
            }
            while(i<nums.size()-2&&nums[i]==nums[i+1]) ++i;//去重
        }
        return result;
    }
};

2.不去重版

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int sum=0,close=INT_MAX,m=0,result=0;
        for(int i=0;i<nums.size()-2;++i){
            m=target-nums[i];
            int l=i+1,r=nums.size()-1;
            while(l<r){
                sum=nums[l]+nums[r];
                if(sum<m){
                    if(m-sum<close){
                        close=m-sum;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    ++l;
                }
                else if(sum>m){
                    if(sum-m<close){
                        close=sum-m;
                        result=nums[i]+nums[l]+nums[r];
                    }
                    --r;
                }
                else{
                    return target;
                }
            }
        }
        return result;
    }
};

猜你喜欢

转载自blog.csdn.net/u011750466/article/details/80221146