给定一个整型数组和一个目标值,找出数组中‘累加和’与‘目标值’最近的三个数的和,假设只存在唯一解
-例子:
Input: array S={-1,2,1,-4},target = 1
Output: 2
刚看到这一题我的想法是3层循环遍历,设置一个变量始终保存 与目标值差绝对值最小的数字,时间复杂度为
O(n3)
,需要考虑降低时间复杂度。**
以前LeetCode上 3Sum的题目,对于求和数组可以先排序,得到一个有序数组。
解题思路
假设数组S有N的元素。
- 首先选定三个元素:i,m = i+1,n = N-1,其中i为基准值从0开始,并设置变量temp保存最接近的数字。
求出S[i] + S[m] + S[n],并将其保存至变量tmp。
if(tmp > target) { n --; } else { m ++; }
每次移动后,比较abs(S[i] + S[m] + S[n] - target) 与abs(tmp - target)的大小,进行替换;当m与n交叉时,结束本轮循环,如果中间遇到S[i] + S[m] + S[n] - target = 0,直接返回target。
然后i++,开启第二轮循环,直至i = N - 2结束,返回tmp。
实现:
int threeSumClosest(vector<int>& nums, int target) {
int tmp = nums[0] + nums[1] + nums[nums.size() -1];
sort(nums.begin(),nums.end());
for(int i = 0;i < nums.size() - 2; i++)
{
int m = i + 1;
int n = nums.size() -1;
while(m < n)
{
int curTmp = nums[i] + nums[m] + nums[n];
if(abs(curTmp - target) < abs(tmp - target))
{
tmp = curTmp;
}
if(curTmp > target)
{
n --;
}
else if(curTmp < target)
{
m ++;
}
else
{
return target;
}
}
}
return tmp;
}