每日一道:最接近的三數之和
阿新 • • 發佈:2021-11-24
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 }