16. 最接近的三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
思路
想到了三数之和的套路,也就是排序+双指针,固定一个在位置i的数后,再[i+1, n-1]范围内进行双指针枚举和cur,用一个全局变量保持三数之和sum,当枚举到的和cur与target的绝对差小于全局和sum与target的绝对差时替代即可。枚举过程中,如果枚举到的和cur小于target时,需要将左指针右移,大于target时,需要将右指针左移,等于直接返回target即可(双指针必备套路,没想到这点所以还是看了题解,唉)。
- 为了避免重复枚举,当前一枚举的数与当前的数相同时,直接跳到下一个数,直到找到不同数为止。
代码
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
int n = nums.size();
int sum = 1e7;
auto update = [&](int cur) {
if(abs(cur - target) < abs(sum - target)) {
sum = cur;
}
};
for(int first = 0; first < n; first++) {
if(first > 0 && nums[first] == nums[first - 1]) {
continue;
}
int left = first + 1, right = n - 1;
while(left < right) {
int cur = nums[first] + nums[left] + nums[right];
if(cur == target) {
return target;
}
update(cur);
if(cur > target) {
int r = right - 1;
while(left < r && nums[r] == nums[r + 1]) {
r--;
}
right = r;
}
else {
int l = left + 1;
while(l < right && nums[l] == nums[l - 1]) {
l++;
}
left = l;
}
}
}
return sum;
}
};