给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数,使它们的和与 target 最接近。
返回这三个数的和。
假定每组输入只存在恰好一个解。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest
例:
输入:nums = [-1,2,1,-4], target = 1 输出:2 解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
解析:双指针,数组从小到大排序,左指针从最左开始,又指针从右开始,然后便利中间的值进行相加,当三数之和小于target时,中间的值右移一个值,当三数之和大于target时,右指针左移一个值,且左指针需要小于右指针。
class Solution(object):
def threeSumClosest(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
nums.sort()
n = len(nums)
nearest = 10 ** 7
for i in range(n - 2):
lo, hi = i + 1, n - 1
# 双指针
while lo < hi:
total = nums[i] + nums[lo] + nums[hi]
if total == target:
return total
elif total < target:
lo += 1
else:
hi -= 1 # 右指针左移
# 取绝对值更小的那个,也更接近
nearest = min(nearest, total, key=lambda x: abs(x - target))
return nearest