結果
| 問題 | 
                            No.198 キャンディー・ボックス2
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2019-08-04 12:51:38 | 
| 言語 | Python3  (3.13.1 + numpy 2.2.1 + scipy 1.14.1)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 29 ms / 1,000 ms | 
| コード長 | 408 bytes | 
| コンパイル時間 | 130 ms | 
| コンパイル使用メモリ | 12,544 KB | 
| 実行使用メモリ | 10,880 KB | 
| 最終ジャッジ日時 | 2024-07-08 01:11:17 | 
| 合計ジャッジ時間 | 1,747 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 26 | 
ソースコード
def median(x):
    N = len(x)
    if N % 2 == 1:
        return x[N//2]
    else:
        return (x[N//2-1] + x[N//2]) // 2
B = int(input())
N = int(input())
C = [int(input()) for _ in range(N)]
C = list(sorted(C))
med = median(C)
if B + sum(C) >= med * N:
    ans = sum(abs(C_i - med) for C_i in C)
    print(ans)
else:
    ave = (B + sum(C)) // N
    ans = sum(abs(C_i - ave) for C_i in C)
    print(ans)