1. 程式人生 > 其它 >每日一道:最接近的三數之和

每日一道:最接近的三數之和

package main

import (
	"fmt"
	"math"
	"sort"
)

/*
給定一個包括 n 個整數的陣列 nums 和 一個目標值 target。找出 nums 中的三個整數,
使得它們的和與 target 最接近。返回這三個數的和。假定每組輸入只存在唯一答案。

輸入:nums = [-1,2,1,-4], target = 1
輸出:2
解釋:與 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
*/
func main() {
	fmt.Println(threeSumClosest1([]int{-1, 2, 1, -4}, 1)) // 2
}
/*
先把前三位數求和,迴圈第一位固定,取第二位和最後一位數,指標收縮
 */
func threeSumClosest1(nums []int, target int) int {
	sort.Ints(nums)
	result := nums[0] + nums[1] + nums[2]
	for i := 0; i < len(nums)-2; i++ {
		m, n := i+1, len(nums)-1
		for m < n {
			// 此處加入比較,當三數之和離目標值更近時,用此三個數之和代替初始的三個數
			sum := nums[i] + nums[m] + nums[n]
			if math.Abs(float64(sum-target)) < math.Abs(float64(result-target)) {
				result = sum
			}
			switch {
			case sum > target:
				n--
			case sum < target:
				m++
			default:
				return sum
			}
		}
	}
	return result
}