题目
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4]
, 和 target = 1
.
与 target
最接近的三个数的和为 2
. (-1 + 2 + 1 = 2
).
分析
这是一道中等难度的题, 是之前三数之和的变种, 条件复杂了一些, 但总体的思路没有变化.
与速度相关的两个关键因素依然是排序算法和去重.
Python的实现上, 调用系统实现的排序即可. C实现可以考虑快速排序, 这样的话第一个因素就解决了. 排序的时间复杂度是O(nlogn)
去重是另一个影响速度的关键因素, 如果没做好, 很可能出现超时. 去重有两个把控的地方, 第一个是外围的循环, 外层循环条件是列表的元素, 出现重复则直接跳过不再经过内层循环. 第二个是内层循环的把关, 内层循环条件是三数之和的第二个数是否小于第三个数(已排序), 在判断与目标值的差距之后, 若当前三数之和res
若小于目标值, 则第二个数的指针向前移动直到没有与当前值重复为止, 若大于, 则第三个数向后移动直到不重复为止.
与目标值的差距用一个变量保存, 初始值为列表前三个数之和(可选其他), 不可以为0或者列表中某个元素的值, 如果不足三个, 则返回. 因为本题默认存在, 因此可不做这个判断.
实现
Python 实现
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
length = len(nums)
if length == 3:
return sum(nums)
nums.sort()
res = sum(nums[:3])
for i in range(length-2):
if i > 0 and nums[i]==nums[i-1]:
continue
l, r = i+1, length-1
while l<r:
tmp = nums[i] + nums[l] + nums[r]
if abs(tmp-target) < abs(res-target):
res = tmp
if tmp < target:
l += 1
elif tmp > target:
r -= 1
else:
return target
return res
效果还不错.