【Leetcode】16. 最接近的三数之和(3Sum Closest)

No16. 最接近的三数之和

题目

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

示例

  • 输入:nums = [-1,2,1,-4], target = 1
  • 输出:2
  • 解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

提示

  • 3 <= nums.length <= 10^3
  • -10^3 <= nums[i] <= 10^3
  • -10^4 <= target <= 10^4

解题代码(Python3)

class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        nums.sort()
        #初始化
        res = 10**4+1
        #用来判断最接近target的数
        def judgeClose(a,b):
            return a if abs(a-target) < abs(b-target) else b
        #子函数 相当于最接近target-value的2个数
        def returnCloseValue(left,right,value):
            #用res来存最接近target的值
            res = num + nums[left] + nums[right]
            while left < right:
                tmp = num + nums[left] + nums[right]
                if tmp > target:
                    right -= 1
                elif tmp < target:
                    left += 1
                #如果等于target 直接返回target
                else:
                    return target
                res = judgeClose(res,tmp)
            return res
        for i, num in enumerate(nums[:-2]):
            #这里大于0是防止对第一个位置判断有误 如果重复则continue
            if i > 0 and num == nums[i-1]:
                continue
            temp = returnCloseValue(i + 1,len(nums) - 1,num)
            if temp == target:
                return target
            else:
                res = judgeClose(res,temp)
        return res

思路:

这题与15题很类似,求三个数a+b+c,使得其和与target最接近,可以先将整个nums进行排序,然后最外层循环固定为a,简化成求另外两个数和target-a的最小差值

复杂度分析:

  • 时间复杂度O(n^2)
  • 空间复杂度O(1)

运行结果:

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/Xiao_Spring/article/details/113731057