@LeetCode最接近的三数之和--3Sum Closest[C++]
问题描述
给定一个包括 n 个整数的数组 nums 和一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1, 2, 1, -4],和 target = 1.
与 target 最接近的三个数的和为2.(-1 + 2 + 1 = 2)
解决方法与复杂度分析
这道题的思路与 3 数之和的思路之和基本一致,需要加上每个结果与目标值的比较,取离目标值最近的那个结果。
在 3 数之和中,我们首先将数组进行排序,然后取 3 个指针,第一个指针指向数组最前端,第二个指针指向第一个指针的下一个元素,第三个指针指向最后一个元素。如果 3 数之和大于目标值,第三个指针前移;否则第二个指针后移。第一个指针依次遍历数组即可。
程序实现
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
int len = nums.size();
int dist = nums[0] + nums[1] + nums[2];
for (int i = 0; i < len - 2; i++) {
int j = i + 1;
int k = len - 1;
while (j < k) {
int sum = nums[i] + nums[j] + nums[k];
if (sum == target) return sum;
if (abs(sum - target) < (dist - target))
dist = sum;
if (sum > target) k--;
else j++;
}
}
return dist;
}
};
@北京·怀柔 2019.3.4