给定一个包括 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;
}
};