假定每组输入只存在唯一答案。
有了上面的假定,可以知道数组中不存在重复元素,和上一题LeetCode第 15 题:三数之和(C++)很像,只是这题target不再固定为0,而且需要寻找最接近的。
不过思路都是一样的,排序之后固定第一个元素,然后使用双指针从两头往中间检索:
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
int distance = INT_MAX;//记录和值与target之间的最小距离
int flag = 1;//代表正
for(int i = 0; i < nums.size()-2; ++i){
int j = i+1, k = nums.size()-1;
while(j < k){
int tmp = nums[i] + nums[j] + nums[k] - target;
if(tmp == 0) return target;//直接返回就行了,已经是最接近
tmp > 0 ? --k : ++j;
if(abs(tmp) < distance){
flag = tmp > 0 ? 1 : -1; //需要记录这个最接近的值是从左边还是右边接近的
distance = abs(tmp);
}
}
}
return target + flag*distance;
}
};