三元组相加获得结果最接近target
3SumClosest
给定一个数组,选择三个元素相加,结果必须为所有三元组中最接近target的值,返回这个三元组的和。
Given an array S of n integers, find three integers in S such that the sum is closest to a given number: target.
Return the sum of the three integers. You may assume that each input would have exactly one solution.
example 1
For example, given array S = [-1, 2, 1, -4], and target = 1.
The sum that is closest to the target is 2. (-1 + 2 + 1 = 2).
input: [-1, 2, 1, -4], 1
output: 2
思路
思路参照三元组相加获得target
只需要将上述文章思路2中换成:第二次循环找到 三元组的和 最接近target的组合即可。
代码
class Solution(object):
def threeSumClosest(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
nums.sort()
closest = nums[0] + nums[1] + nums[2]
for i in range(len(nums)):
j, k = i + 1, len(nums) - 1
while j < k:
value = nums[i] + nums[j] + nums[k]
closest = value if abs(target - value) < abs(target - closest) else closest
if value == target:
return target
elif value > target:
k -= 1
else:
j += 1
return closest
本题以及其它leetcode题目代码github地址: github地址