16. 最接近的三数之和(medium)

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

  • show the code:
class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        nums.sort()
        n,res = len(nums),nums[0]+nums[1]+nums[2]
        for i in range(n):
            l,r = i+1,n-1
            while l < r:
                s = nums[i]+nums[l]+nums[r]
                if abs(s-target) < abs(res-target):
                    res = s
                if s < target:
                    l += 1
                elif s > target:
                    r -= 1
                else:
                    return target
        return res
  • 此题和上一题3sum非常相似,可以说是它的变种。

  • 同样的,方法是双指针法,先对数组进行排序,加快我们运算的速度,然后确定一个数,另外两个数分别处于左边和最右边。如果比target小则左指针右移,大则右指针左移。由于题目中强调过了:假定每组输入只存在唯一答案。所以我们无需考虑重复的数。

  • 这里唯一和3sum处理方式不同的是,我们不存储中间的结果,而是直接更新我们的res,这个res最开始确定成前三个数之和,后面如果出现比它更接近target的数,再更新res。最终就可以得到最接近target的三数之和。

时间复杂度:n2,空间复杂度:1

你可能感兴趣的:(16. 最接近的三数之和(medium))