結果

問題 No.198 キャンディー・ボックス２
ユーザー fmhr
提出日時 2015-07-29 22:41:38
言語 Go
(1.23.4)
結果
AC  
実行時間 2 ms / 1,000 ms
コード長 853 bytes
コンパイル時間 13,642 ms
コンパイル使用メモリ 226,572 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-16 09:05:08
合計ジャッジ時間 12,705 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

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
}
0