给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法:
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
nums.sort()
l = 0
r = 0
best_sum = 0
best_dist = float('inf')
for i in range(len(nums) - 2):
l = i + 1
r = len(nums) - 1
while l < r:
tmp_sum = nums[i] + nums[l] + nums[r]
if abs(tmp_sum - target) < best_dist:
best_dist, best_sum = abs(tmp_sum - target), tmp_sum
if tmp_sum < target:
l += 1
elif tmp_sum > target:
r = r - 1
else:
return target
return best_sum