分类:algorithm| 发布时间:2016-07-03 23:50:00
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
这个问题跟 3Sum 很接近,因此解法的思路也基本一样。 不过不用记录所有可能的情况,只用找出最接近 target 的和就行。
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
int minDiff = INT_MAX;
int ret = 0;
for (int i = 0; i < nums.size(); i++) {
int front = i + 1, end = nums.size() - 1;
while (front < end) {
int sum = nums[i] + nums[front] + nums[end];
if (sum > target) {
--end;
} else if (sum < target) {
++front;
} else {
return target;
}
int diff = abs(sum - target);
if (minDiff > diff) {
ret = sum;
minDiff = diff;
}
}
}
return ret;
}
};