結果

問題 No.198 キャンディー・ボックス２
コンテスト
ユーザー hiragn
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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()
0