Leetcode: 最接近的三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
先排序,然后从两端开始向中间搜索,i为遍历后的数字,R表示最右端数字,L表示nums[i + 1],我们考虑这三个数与target的差值,若是大于0,则我们把R挪一位(R-=1),若是小于0,则将L挪一位(L+=1),等于0直接返回target, 同时每次操作时把三位数与target的差值放入列表中,然后把 i 挪一位(i +=1),再重复上面的操作。最后我们会得到三个数与target差值的列表,再对其排序,我们可以找到唯一的一组相邻的两个数数,比较这两个数的绝对值,返回绝对值小的那个数与target的和。
class Solution:
def threeSumClosest(self, nums, target):
nums.sort()
res_sum = nums[0] + nums[1] + nums[2]
res_abs = abs(nums[0]+nums[1]+nums[2]-target)
for i in range(len(nums)-2):
L = i + 1
R = len(nums) - 1
while L < R:
three_sum = nums[i] + nums[L] + nums[R]
abs_sub = abs(three_sum - target)
if abs_sub == 0:
return target
elif abs_sub < res_abs:
res_abs = abs_sub
res_sum = three_sum
elif three_sum > target:
R -= 1
else:
L += 1
return res_sum