#40185 (Go) No.198 キャンディー・ボックス２

• 問題: No.198 キャンディー・ボックス２
• ユーザー: fmhr
• 提出日時: 2015-07-29 22:41:38
• 言語: Go (1.22.1)
• 結果: AC
• 実行時間: 1 ms / 1,000 ms
• コード長: 853 bytes
• コンパイル時間: 13,406 ms
• コンパイル使用メモリ: 235,604 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-28 01:10:07
• 合計ジャッジ時間: 10,508 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	1 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	AC	1 ms; 5,376 KB
testcase_15	AC	1 ms; 5,376 KB
testcase_16	AC	1 ms; 5,376 KB
testcase_17	AC	1 ms; 5,376 KB
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	1 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	1 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	1 ms; 5,376 KB
testcase_29	AC	1 ms; 5,376 KB

ソースコード

package main

import (
	"fmt"
	"math"
)

func main() {
	solve()
}

var B, N, ans int
var C [10]int

func solve() {
	fmt.Scan(&B, &N)
	for i := 0; i < N; i++ {
		fmt.Scan(&C[i])
	}
	ans = math.MaxInt64
	lo := 0
	hi := math.MaxInt64 / 30
	for hi-lo > 10 {
		loMid := (lo + lo + hi) / 3
		hiMid := (lo + hi + hi) / 3
		if calc(loMid) <= calc(hiMid) {
			hi = hiMid
		} else {
			lo = loMid
		}
	}
	for i := max(0, lo-100); i < lo+100; i++ {
		calc(i)
	}
	fmt.Println(ans)
}

func calc(a int) int {
	rest := B
	temp := 0
	for i := 0; i < N; i++ {
		temp += abs(a - C[i])
		rest += C[i] - a
	}
	if rest < 0 {
		return math.MaxInt64 / 3
	}
	ans = min(temp, ans)
	return temp
}

func abs(a int) int {
	if a < 0 {
		a *= -1
	}
	return a
}

func min(a, b int) int {
	if a > b {
		a = b
	}
	return a
}

func max(a, b int) int {
	if a < b {
		a = b
	}
	return a
}