leetcode-16-3sum-closest

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

1
2
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

Solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
class Solution(object):
def threeSumClosest(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
nums.sort()
res = sum(nums[:3])
for i in range(len(nums) - 2):
l = i + 1
r = len(nums) - 1
while l < r:
three = nums[i] + nums[l] + nums[r]
if abs(three - target) < abs(res - target):
res = three
if three > target:
r -= 1
elif three < target:
l += 1
else:
return res
return res