16. 最接近的三数之和
- 题目描述
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2). - 代码实现
class Solution(object):
def threeSumClosest(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
lens = len(nums)
nums.sort()
res = float("inf")
for k in range(lens - 2):
l = k+1
r = lens-1
if k>0 and nums[k] == nums[k-1]:
continue
while l<r:
var = nums[k]+nums[l]+nums[r]
if var == target:
return target
if abs(var - target)< abs(res - target):
res = var
if var > target:
r -= 1
if var < target:
l += 1
return res
刚开始的时候代码写成了:
class Solution(object):
def threeSumClosest(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
lens = len(nums)
nums.sort()
res = float("inf")
for k in range(lens - 2):
l = k+1
r = lens-1
if k>0 and nums[k] == nums[k-1]:
continue
var = nums[k]+nums[l]+nums[r]
while l<r:
if var == target:
return target
if abs(var - target)< abs(res - target):
res = var
if var > target:
r -= 1
if var < target:
l += 1
return res
var = nums[k]+nums[l]+nums[r] : 应该放在while判断里。