結果

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

ソースコード

diff # 

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