結果
問題 | No.198 キャンディー・ボックス2 |
ユーザー |
![]() |
提出日時 | 2023-04-21 11:59:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 41 ms / 1,000 ms |
コード長 | 902 bytes |
コンパイル時間 | 688 ms |
コンパイル使用メモリ | 82,276 KB |
実行使用メモリ | 54,008 KB |
最終ジャッジ日時 | 2024-11-06 08:10:49 |
合計ジャッジ時間 | 2,482 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
# medianにすればいいか # WA出た # median総数に足りない場合はmedianを下げる必要があるな B = int(input()) N = int(input()) C = [] for i in range(N): C.append(int(input())) C.sort() sumC = sum(C) from math import floor, ceil if N%2 == 1: median = C[N//2] median = min(median, floor((sumC+B)/N)) ans = 0 for i in range(N): ans += abs(C[i]-median) #print(median, ans) else: median_low = floor((C[N//2-1]+C[N//2])/2) median_low = min(median_low, floor((sumC+B)/N)) ans_low = 0 for i in range(N): ans_low += abs(C[i]-median_low) #print(median_low, ans_low) median_high = ceil((C[N//2-1]+C[N//2])/2) median_high = min(median_high, floor((sumC+B)/N)) ans_high = 0 for i in range(N): ans_high += abs(C[i]-median_high) #print(median_high, ans_high) ans = min(ans_low, ans_high) print(ans)