[LeetCode]16. 最接近的三数之和
题目
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
示例 1
输入:nums = [-1,2,1,-4], target = 1
输出:2
与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
解析
- 分析
-
跟15.三数之和类似
-
abs(target - current)
这个值越少说明越接近
- 算法流程
-
排序 - 为了能使用双指针
-
遍历
- 固定一位(i), 左右指针指向下一位与最后一位(
left = i + 1
,right = n - 1
) - 按条件遍历左右指针
- i + left + right == 0 : 当前和
target
重合了,直接返回 - i + left + right < 0 : 结果偏少,需要增加结果 left++(数组排序后从左往右依次增大)
- i + left + right > 0 : 结果偏大,需要减少结果 right--
- i + left + right == 0 : 当前和
- 固定一位(i), 左右指针指向下一位与最后一位(
源码
from typing import List
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
nums.sort()
length = len(nums)
res: int = nums[0] + nums[1] + nums[2]
for i in range(length-2):
left = i + 1
right = length - 1
while left < right:
total = (nums[i] + nums[left] + nums[right])
if abs(target - total) < abs(target - res):
res = total
if target == total:
return total
elif total > target:
right = right - 1
else:
left = left + 1
return res