给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
- show the code:
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
nums.sort()
n,res = len(nums),nums[0]+nums[1]+nums[2]
for i in range(n):
l,r = i+1,n-1
while l < r:
s = nums[i]+nums[l]+nums[r]
if abs(s-target) < abs(res-target):
res = s
if s < target:
l += 1
elif s > target:
r -= 1
else:
return target
return res
此题和上一题3sum非常相似,可以说是它的变种。
同样的,方法是双指针法,先对数组进行排序,加快我们运算的速度,然后确定一个数,另外两个数分别处于左边和最右边。如果比target小则左指针右移,大则右指针左移。由于题目中强调过了:假定每组输入只存在唯一答案。所以我们无需考虑重复的数。
这里唯一和3sum处理方式不同的是,我们不存储中间的结果,而是直接更新我们的
res
,这个res
最开始确定成前三个数之和,后面如果出现比它更接近target
的数,再更新res
。最终就可以得到最接近target
的三数之和。