1. 题目来源
链接:16. 最接近的三数之和
2. 题目解析
方法一:枚举+双指针
前导题:
和前导题思路完全一样。且本题保证了仅有一组答案,还不需要考虑去重的事情了。
双指针思路:
- 先枚举一个数,记为指针
i
。再进行双指针,j
在i
的后一位,k
在数组末尾。 - 排序后。双指针找出使得:
num[i]+num[j]+num[k] >= target
成立的最小的k
,那么此时num[i]+num[j]+num[k - 1] < target
必然成立,当然在此j-1
和k
不能相同。在两种情况中选择一个最小值来更新答案即可。 - 所以,针对每一个
num[i]
就可以用 O ( n ) O(n) O(n) 的时间找到合法的j, k
。 故总的时间复杂度就是 O ( n 2 ) O(n^2) O(n2) 的。
注意:
- 计算过程中,首先要存三数之和,它是要最终返回的答案,及其与
target
的差值,要用它来不断更新答案,采用pair<int, int> res(INT_MAX, INT_MAX);
比较方便一点。 - 在更新答案差值的过程中,取两个差值的
min
来更新,但可能涉及到负数,如果这个target
很大,三数之和记为num[i]+num[j]+num[k]=s
,但针对于k
来讲,这个三数之和s
是不应大于target
,那么这个距离差值就变成负数,则不合法。在k - 1 > j
的情况下,针对num[i]+num[j]+num[k - 1] < target
来讲是必然成立的,这里就不需要加abs
了。
时间复杂度:
O
(
n
2
)
O(n^2)
O(n2)
空间复杂度:
O
(
1
)
O(1)
O(1)
代码:
class Solution {
public:
int threeSumClosest(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
pair<int, int> res(INT_MAX, INT_MAX);
for (int i = 0; i < nums.size(); i ++) {
for (int j = i + 1, k = nums.size() - 1; j < k; j ++) {
while (k - 1 > j && nums[i] + nums[j] + nums[k - 1] >= target) k --;
int s = nums[i] + nums[j] + nums[k];
res = min(res, make_pair(abs(s - target), s));
if (k - 1 > j) {
s = nums[i] + nums[j] + nums[k - 1];
res = min(res, make_pair(target - s, s));
}
}
}
return res.second;
}
};