結果

問題 No.198 キャンディー・ボックス２
コンテスト
ユーザー 山本信二
提出日時 2022-05-25 14:23:39
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 384 ms / 1,000 ms
コード長 413 bytes
コンパイル時間 124 ms
コンパイル使用メモリ 12,160 KB
実行使用メモリ 31,952 KB
最終ジャッジ日時 2025-11-26 11:46:22
合計ジャッジ時間 12,748 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

import numpy as np

data = np.fromstring(read(), np.int64, sep=' ')

B = data[0]
N = data[1]
C = data[2:]

INF = 10 ** 18

def f(x):
    if C.sum() + B < x * N:
        return INF
    return np.abs(C-x).sum()

cand = C.tolist() + [(B + C.sum())//N]

answer = min(f(x) for x in cand)
print(answer)
0