leetcode 16:最接近的三数之和

Problem:

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum-closest

Answer:

1、暴力解法:时间复杂度O(n^3),超出时间限制。

 

2、O(n^2)的解法:

先将数组排序,对于每一个i从0到n - 1，令left = i + 1，right = size - 1，为了寻找更接近target的三数之和，

若三数之和大于target则让right - 1，若三数之和等于target则返回target，若三数之和小于target则让left + 1。

 

设排序后的数组arr为a, b, c, d, e, f。

令arr[i] = a, arr[left] = b, arr[right] = f。

若a + b + f < target，则令left + 1，即arr[left] = c，

b不再有与小于f的数（如e）组合的可能，但由于e < f，a + b + e < a + b + f，离target更远，因而不必考虑。

 

Code:

class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        nums_len = len(nums)
        nums.sort()
        min_dist = float('inf')
        for i in range(nums_len):
            left = i + 1
            right = nums_len - 1
            while left < right:
                sum_ = nums[i] + nums[left] + nums[right]
                dist = abs(target - sum_)
                if dist < min_dist:
                    min_dist = dist
                    ans = sum_
                if sum_ == target:
                    return target
                elif sum_ > target:
                    right -= 1
                else:
                    left += 1
        return ans