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

• 問題: No.198 キャンディー・ボックス２
• ユーザー: fmhr
• 提出日時: 2015-07-29 22:39:30
• 言語: Go (1.22.1)
• 結果: WA
• 実行時間: -
• コード長: 852 bytes
• コンパイル時間: 12,550 ms
• コンパイル使用メモリ: 233,888 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-19 03:13:46
• 合計ジャッジ時間: 13,801 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	WA	-
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
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	WA	-
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	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-

ソースコード

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 / 3
	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
}