16. 最接近的三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
示例:
输入: nums = [-1,2,1,-4], target = 1
输出: 2
解释: 与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
提示:
- 3 < = n u m s . l e n g t h < = 1 0 3 3 <= nums.length <= 10^3 3<=nums.length<=103
- − 1 0 3 < = n u m s [ i ] < = 1 0 3 -10^3 <= nums[i] <= 10^3 −103<=nums[i]<=103
- − 1 0 4 < = t a r g e t < = 1 0 4 -10^4 <= target <= 10^4 −104<=target<=104
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
双指针
此题是15. 三数之和的升级,区别在于将求定值变成了求最接近的值。思路还是一样的,不同的是维护一个变量记录当前确定的三个数的和于目标值的差值,并更新到最小,直至全部遍历一遍。于是在【三数之和】的代码基础上修改。而且【三数之和】输出的是三元组,还需要去重,这里只需要求值,可以删去不少代码。
此题用哈希表反而代码臃肿,不如只用双指针来的干脆,不得不说在保持有序这一块哈希表并不是很好用。
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
if not nums or len(nums)<3: return 0
if len(nums) == 3: return sum(nums)
ans = float('inf')
nums.sort()
for i in range(len(nums)):
if i > 0 and nums[i] == nums[i-1]: continue
left, right = i + 1, len(nums) - 1
while left < right:
if target == nums[i] + nums[left] + nums[right]: return target
if abs(target - (nums[i] + nums[left] + nums[right])) < abs(target - ans):
ans = nums[i] + nums[left] + nums[right]
if target > nums[i] + nums[left] + nums[right]:
left += 1
else:
right -= 1
return ans