力扣做题16-最接近的三数之和

给定一个包括 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

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

解题思路
首先将数组升序排列，然后依次取数组的第 i 个数作为选取的第一个数，再从其之后的数里面，用左右指针的方法分别选出剩余的两个数。计算这三个数与目标的差值，并用一个变量进行记录更新。最终，若差值等于 0，则直接返回目标；否则，返回目标与差值的和。

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

优化：选择第一个数字时进行去重。

代码

class Solution(object):
    def threeSumClosest(self, nums, target):
        nums.sort()
        min_diff = float('inf')
        for i in range(len(nums) - 2):
            if i >= 1 and nums[i] == nums[i-1]:  # 跳过重复元素
                continue
            left = i + 1  # 左指针
            right = len(nums) - 1  # 右指针
            while left < right:
                cur_diff = nums[i] + nums[left] + nums[right] - target  # 当前和与 target 的差值
                if cur_diff == 0:  # 等于 target
                    return target
                if cur_diff > 0:  # 大于 target
                    right -= 1
                else:  # 小于 target
                    left += 1
                min_diff = cur_diff if abs(cur_diff) < abs(min_diff) else min_diff  # 更新最小差值
        return target + min_diff  # 结果加上差值