結果
問題 | No.198 キャンディー・ボックス2 |
ユーザー |
![]() |
提出日時 | 2022-11-19 00:13:27 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 409 ms / 1,000 ms |
コード長 | 431 bytes |
コンパイル時間 | 691 ms |
コンパイル使用メモリ | 12,160 KB |
実行使用メモリ | 32,408 KB |
最終ジャッジ日時 | 2025-03-15 00:34:50 |
合計ジャッジ時間 | 13,690 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
import numpy as np def main(): b = int(input()) n = int(input()) c = np.array([int(input()) for _ in range(n)], dtype='int64') # bが小さいと中央値np.median(c)を実現できない。 # n * x = sum(c) + bをみたすxと中央値の小さい方を作る x = min(int(np.median(c)), (np.sum(c) + b) // n) ans = sum([abs(i - x) for i in c]) print(ans) if __name__ == "__main__": main()