题目
给定一个包括 n
个整数的数组 nums
和 一个目标值 target
。找出 nums
中的三个整数,使得它们的和与 target
最接近。返回这三个数的和。假定每组输入只存在唯一答案。
示例:
输入:nums = [-1,2,1,-4], target = 1 输出:2 解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
解题
方法一:排序+双指针
这道题的总体思路是和leetcode-15:三数之和是一样的,只是略微的进行调整。
官方的解题链接其实不用看,看完LC 三数之和的直接看代码就明白了。
官方的解答(比较冗长,直接看最下面简洁的版本就行了)
class Solution: def threeSumClosest(self, nums: List[int], target: int) -> int: nums.sort() n = len(nums) best = 10**7 # 根据差值的绝对值来更新答案 def update(cur): nonlocal best if abs(cur - target) < abs(best - target): best = cur # 枚举 a for i in range(n): # 保证和上一次枚举的元素不相等 if i > 0 and nums[i] == nums[i - 1]: continue # 使用双指针枚举 b 和 c j, k = i + 1, n - 1 while j < k: s = nums[i] + nums[j] + nums[k] # 如果和为 target 直接返回答案 if s == target: return target update(s) if s > target: # 如果和大于 target,移动 c 对应的指针 k0 = k - 1 # 移动到下一个不相等的元素 while j < k0 and nums[k0] == nums[k]: k0 -= 1 k = k0 else: # 如果和小于 target,移动 b 对应的指针 j0 = j + 1 # 移动到下一个不相等的元素 while j0 < k and nums[j0] == nums[j]: j0 += 1 j = j0 return best
经过一些调整,看起来更加简洁和舒适了。
代码:
class Solution: def threeSumClosest(self, nums: List[int], target: int) -> int: nums.sort() n = len(nums) best = 10**7 for i in range(n): if i > 0 and nums[i] == nums[i - 1]: continue L, R = i + 1, n - 1 while L < R: s = nums[i] + nums[L] + nums[R] if s == target: return target if abs(s - target) < abs(best - target): best = s if s > target: R = R - 1 else: L = L + 1 return best