题目描述
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
算法设计
用三个指针i,j,k以i=0, j=i+1, k=nums.length()-1为起点扫描,当三个数的和小于最小值时,更新sum。时间复杂度O(n^3)
C++代码
class Solution {
public:
static bool cmp(const int a, const int b){
return a<b;
}
int threeSumClosest(vector<int>& nums, int target) {
int len = nums.size(), minus, sum=0;
int num_i[len];
if(len<3){
return 0;
}
for(int i=0;i<len;i++){
num_i[i] = nums[i];
}
sort(num_i, num_i+len, cmp);
sum = num_i[0] + num_i[1] + num_i[2];
minus = abs(target - sum);
for(int i=0;i<len;i++){
for(int j=i+1;j<len;j++){
for(int k=len-1;k>=j+1;k--){
int tmpsum = num_i[i] + num_i[j] + num_i[k];
int delta = abs(target - tmpsum);
if(tmpsum == target){
sum = tmpsum;
return sum;
}
if(delta > minus){
continue;
}
else{
minus = delta;
sum = num_i[i] + num_i[j] + num_i[k];
}
}
}
}
return sum;
}
};
优化算法
将数组排序,第二层循环使用双指针,一个指向i+1,一个指向末尾。计算三个数的和sum,如果target-sum = delta<minus则更新sum。
当sum<target时,将头指针向后移动;
当sum>target时,将尾指针向前移动;
当尾指针和头指针重叠时,结束。
优化后的时间复杂度为o(n^2)。
优化C++代码
class Solution {
public:
static bool cmp(const int a, const int b){
return a<b;
}
int threeSumClosest(vector<int>& nums, int target) {
int len = nums.size(), minus, sum=0;
int num_i[len];
if(len<3){
return 0;
}
for(int i=0;i<len;i++){
num_i[i] = nums[i];
}
sort(num_i, num_i+len, cmp);
sum = num_i[0] + num_i[1] + num_i[2];
minus = abs(target - sum);
for(int i=0;i<len;i++){
int j = i+1, k = len-1;
while(j < k && j < len && k < len){
int tmpsum = num_i[i] + num_i[j] + num_i[k];
int delta = abs(target - tmpsum);
if(delta < minus){
minus = delta;
sum = tmpsum;
}
if(tmpsum > target){
k--;
}
else if(tmpsum < target){
j++;
}
else{
return sum;
}
}
}
return sum;
}
};