LeetCode每日一题14. 最接近的三数之和

LeetCode

题目

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

例如，给定数组 nums = [-1，2，1，-4], 和 target = 1.

与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

---

分析

这是一道中等难度的题, 是之前三数之和的变种, 条件复杂了一些, 但总体的思路没有变化.
与速度相关的两个关键因素依然是排序算法和去重.

Python的实现上, 调用系统实现的排序即可. C实现可以考虑快速排序, 这样的话第一个因素就解决了. 排序的时间复杂度是O(nlogn)

去重是另一个影响速度的关键因素, 如果没做好, 很可能出现超时. 去重有两个把控的地方, 第一个是外围的循环, 外层循环条件是列表的元素, 出现重复则直接跳过不再经过内层循环. 第二个是内层循环的把关, 内层循环条件是三数之和的第二个数是否小于第三个数(已排序), 在判断与目标值的差距之后, 若当前三数之和res若小于目标值, 则第二个数的指针向前移动直到没有与当前值重复为止, 若大于, 则第三个数向后移动直到不重复为止.

与目标值的差距用一个变量保存, 初始值为列表前三个数之和(可选其他), 不可以为0或者列表中某个元素的值, 如果不足三个, 则返回. 因为本题默认存在, 因此可不做这个判断.

---

实现

Python 实现

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

结果

效果还不错.