16. 最接近的三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
示例:
输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
提示:
3 <= nums.length <= 10^3
-10^3 <= nums[i] <= 10^3
-10^4 <= target <= 10^4
M = float("inf") # 三个数的和初步设置为最大值
lens = len(nums)
nums.sort() # 排序
for i in range(lens):
if i>0 and nums[i]==nums[i-1]: #如果nums[i]的值相等,则跳过
continue
left = i + 1
right = lens -1
while left < right:
ans = nums[i] + nums[left] + nums[right]
if target == ans:
return target
#如果差值更接近,则保存差值更接近的三个数的和
if abs(ans - target) < abs(M-target):
M = ans
if target > ans: #target太大,说明左边的值需要往右边移动来靠近target
while left<right and nums[left]==nums[left+1]:
left += 1
left += 1
elif target < ans: #target太小,说明右边的值需要往左边移动变小来靠近target
while left<right and nums[right]==nums[right-1]:
right -= 1
right -= 1
return M