LeetCode-16

最新推荐文章于 2024-07-22 17:10:05 发布

楚仙子

最新推荐文章于 2024-07-22 17:10:05 发布

阅读量171

点赞数

分类专栏： LeetCode 文章标签： leetcode 数据结构 python 算法指针

本文链接：https://blog.csdn.net/weixin_43901733/article/details/109035103

版权

LeetCode 专栏收录该内容

42 篇文章 0 订阅

订阅专栏

16. 最接近的三数之和

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

Example

input
nums = [-1,2,1,-4], target = 1
output
2

Note

样例中与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
$3 <= nums.length <= 10^3$
$10^3 <= nums[i] <= 10^3$
$10^4 <= target <= 10^4$

思路

大体思路和“三数之和”一样，双指针+排序
可以在每次双指针之前就进行判断，具体见代码

代码如下

class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        n = len(nums)
        nums.sort()
        ans = nums[0] + nums[1] + nums[2]
        for i in range(n-2):
            if i > 0 and nums[i] == nums[i-1]:
                continue
            l, r = i+1, n-1
            max_num = nums[i] + nums[n-1] + nums[n-2]
            min_num = nums[i] + nums[i+1] + nums[i+2]
            if max_num < target:
                if abs(max_num - target) < abs(ans - target):
                    ans = max_num
                continue
            if min_num > target:
                if abs(min_num - target) < abs(ans - target):
                    ans = min_num
                break
            while l < r:
                res = nums[i] + nums[l] + nums[r]
                if abs(res - target) < abs(ans - target):
                    ans = res
                if res == target:
                    return ans
                elif res > target:
                    r -= 1
                else:
                    l += 1
        return ans