結果

問題 No.198 キャンディー・ボックス２
ユーザー 山本信二山本信二
提出日時 2022-05-25 14:23:39
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 521 ms
43,432 KB
testcase_01 AC 446 ms
43,668 KB
testcase_02 AC 460 ms
43,468 KB
testcase_03 AC 455 ms
43,432 KB
testcase_04 AC 453 ms
43,820 KB
testcase_05 AC 453 ms
43,564 KB
testcase_06 AC 450 ms
43,688 KB
testcase_07 AC 445 ms
43,436 KB
testcase_08 AC 454 ms
43,308 KB
testcase_09 AC 454 ms
43,568 KB
testcase_10 AC 451 ms
43,432 KB
testcase_11 AC 455 ms
43,792 KB
testcase_12 AC 454 ms
43,412 KB
testcase_13 AC 453 ms
43,496 KB
testcase_14 AC 443 ms
43,412 KB
testcase_15 AC 457 ms
43,556 KB
testcase_16 AC 449 ms
43,304 KB
testcase_17 AC 452 ms
43,692 KB
testcase_18 AC 458 ms
43,696 KB
testcase_19 AC 452 ms
43,564 KB
testcase_20 AC 456 ms
43,304 KB
testcase_21 AC 456 ms
43,692 KB
testcase_22 AC 453 ms
43,696 KB
testcase_23 AC 454 ms
43,432 KB
testcase_24 AC 458 ms
43,432 KB
testcase_25 AC 457 ms
43,820 KB
testcase_26 AC 455 ms
43,796 KB
testcase_27 AC 461 ms
43,816 KB
testcase_28 AC 456 ms
43,692 KB
testcase_29 AC 460 ms
43,820 KB
権限があれば一括ダウンロードができます

ソースコード

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