题目描述:给出一个
target
找出数组中3个元素和最接近target的和,并返回其和值。
题目链接:16. 3Sum Closest
这个题跟3Sum很像,只不过要求target最接近,那么我在每次更新的时候就保留和值就可以了。
代码如下
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
k = target
if len(nums) < 3:
return 0
ans = 0
nums.sort()
curr_diff = (1<<31) - 1 #首先放到最大
for i in range(0,len(nums)-2): #到最后两个就停下
fix = k - nums[i] #需要fix的数 如果和前面一个数字相等 则continue k 要找到与fix最接近的数就是说ans
if (i > 0 and nums[i] == nums[i-1]): #去重
continue
l = i + 1 #fix 数的下一个数 开始查询[l,r] 两个数相加和为 fix,注意去重。
r = len(nums) - 1 #需要fix的数的右边数组的右边界
while(l < r): #当左边界小于右边界时
curr_sum = nums[l] + nums[r]
if curr_diff > abs(curr_sum-fix):#如果这个绝对值小于 curr_sum 就更新值
curr_diff = abs(curr_sum-fix)
ans = nums[l] + nums[r] + nums[i]
if (curr_sum > fix): #太大了则右边往左
r -= 1
elif (curr_sum < fix): # 太小了往右
l += 1
else:
return target
return ans