16_3sum_closest

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

代码参考：

package main

import (
    "fmt"
    "sort"
)

func main() {
    fmt.Println(threeSumClosest1([]int{-1, 2, 1, -4}, 1))                // 2
    fmt.Println(threeSumClosest([]int{0, 2, 1, -3}, 1))                  // 0
    fmt.Println(threeSumClosest([]int{1, 2, 4, 8, 16, 32, 64, 128}, 84)) // 84
}

// 同样也是指针法
// 和 15 一样，排序预处理能知道双指针移动的方向，记录最小 abs
func threeSumClosest(nums []int, target int) int {
    sort.Ints(nums)
    n := len(nums)
    minAbs := 1<<31 - 1
    minSum := 0

    for i, num := range nums {
        if i > 0 && nums[i] == nums[i-1] { // 优化：可选的去重
            continue
        }

        l, r := i+1, n-1
        for l < r {
            sum := num + nums[l] + nums[r]
            if abs(target-sum) < minAbs {
                minAbs = abs(target - sum)
                minSum = sum
            }
            switch {
            case sum < target:
                l++
            case sum > target:
                r--
            default:
                return target
            }
        }
    }
    return minSum
}

// 暴力三层遍历，求解最接近的和
func threeSumClosest1(nums []int, target int) int {
    sort.Ints(nums)
    n := len(nums)
    minAbs := abs(nums[0] + nums[1] + nums[2] - target)
    minSum := 0
    for i := 0; i < n-2; i++ {
        for j := i + 1; j < n-1; j++ {
            for k := j + 1; k < n; k++ {
                sum := nums[i] + nums[j] + nums[k]
                if abs(sum-target) <= minAbs {
                    minSum = sum
                    minAbs = abs(sum - target)
                }
            }
        }
    }
    return minSum
}