持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第9天,点击查看活动详情
一 描述
给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数,使它们的和与 target 最接近。
返回这三个数的和。
假定每组输入只存在恰好一个解。
示例 1:
输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
示例 2:
输入:nums = [0,0,0], target = 1
输出:0
提示:
- 3 <= nums.length <= 1000
- -1000 <= nums[i] <= 1000
- -10^4 <= target <= 10^4
二 分析
- 对数组进行排序
- 初始化三元素之和 total = nums[0]+nums[1]+nums[2],由于target不确定,total不能设定为0或-1
- 双指针流程:
- 若nums[i] > target,退出循环:排序之后,后面的元素(nums[i+1]和nums[i+2])都大于等于nums[i],因此结果不可能等于target
- 令左指针left = i+1,右指针right = length-1,执行循环,判断选取的三元素之和total与target之间的关系,移动左右指针,以寻找最优结果。
- cha用来存储total - target的最小绝对值(当前状态下),完成所有遍历后,最小绝对值所对应的元素和为最终结果
- 若 abs(total-target) < cha,更新cha和res
- 当total - target > 0时,right = right - 1
- 当total - target < 0时,left = left + 1
- 当total - target = 0时,所取三元素之和等于target,可以直接返回结果target
- 遍历所有完成,返回res
三 答案
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
nums.sort()
length = len(nums)
total = nums[0]+nums[1]+nums[2]
cha = abs(total - target)
res = total
for i in range(length-2):#第一层循环,固定其中一个元素
if nums[i] > target:
break
left = i+1
right = length-1
while(left < right):#使用双指针
total = nums[i]+nums[left]+nums[right]
if abs(total-target) < cha:
cha = abs(total - target)
res = total
if total - target > 0:
right = right - 1
elif total - target < 0:
left = left + 1
else:
return target
return res