LeetCode 16.最接近的三数之和

题目

16.最接近的三数之和

给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数，使它们的和与 target 最接近。
返回这三个数的和。
假定每组输入只存在恰好一个解。

示例 1：

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

示例 2：

输入：nums = [0,0,0], target = 1
输出：0

提示：

• 3 <= nums.length <= 1000
• -1000 <= nums[i] <= 1000
• -10⁴ <= target <= 10⁴

思路

算法：
排序+双指针算法 O(n²)

• 1、枚举每个数，表示该数 num[i] 已被确定，在排序后的情况下，通过双指针l，r分别从左边 l = i + 1 和右边n - 1往中间靠拢，找到sum = nums[i] + nums[l] + nums[r]所有情况中最接近target的sum，更新ans
• 2、在找的过程中，假设sum = nums[i] + nums[l] + nums[r]
  若sum > target，则r往左走，使sum变小，更接近target；
  若sum < target，则l往右走，使sum变大，更接近target；
  若sum == target，则表示找到了与num[i]搭配的组合num[l]和num[r]，直接返回；
• 3、判重处理
  确定好nums[i]时，l 需要从i + 1开始
  当nums[i] == nums[i - 1]，表示当前确定好的数与上一个一样，需要直continue

时间复杂度：总的时间复杂度为 O(n²) .

代码

C++代码：

class Solution {
    public int threeSumClosest(int[] nums, int target) {
        int n = nums.length;
        int ans = 0x3f3f3f3f;
        Arrays.sort(nums);
        for(int i = 0;i < n;i ++)
        {
            if(i != 0 && nums[i] == nums[i - 1]) continue;

            int l = i + 1,r = n - 1;

            while(l < r)
            {
                int sum = nums[i] + nums[l] + nums[r];
                if(Math.abs(sum - target) < Math.abs(ans - target)) ans = sum;
                if(sum > target)
                {
                    r -- ;
                    continue;
                }

                if(sum < target)
                {
                    l ++ ;
                    continue;
                }

                return target;
            }
        }
        return ans;
    }
}

python3代码：

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