結果

問題 No.198 キャンディー・ボックス２
ユーザー FromBooska
提出日時 2023-10-17 12:59:02
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,231 bytes
コンパイル時間 625 ms
コンパイル使用メモリ 82,596 KB
実行使用メモリ 70,568 KB
最終ジャッジ日時 2024-09-17 11:01:54
合計ジャッジ時間 2,998 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 RE * 1
other AC * 24 RE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 二分探索に見えるが、単調ではなく2次曲線となるので三分探索

B = int(input())
N = int(input())
C = []
for i in range(N):
    C.append(int(input()))

# 三分探索ではreturn 0 or 1ではなくコストが必要
def check(X):
    count = 0
    for c in C:
        count += abs(c-X)
    if count <= X:
        return count
    else:
        return count
    
#for i in range(0, 10):
#    print(i, check(i))

# 谷、つまり1極値を求める三分探索での解き方
# https://roiti46.hatenablog.com/entry/2015/04/29/yukicoder_No.198_%E3%82%AD%E3%83%A3%E3%83%B3%E3%83%87%E3%82%A3%E3%83%BC%E3%83%BB%E3%83%9C%E3%83%83%E3%82%AF%E3%82%B9%EF%BC%92
# https://qiita.com/ganyariya/items/1553ff2bf8d6d7789127

# 範囲に注意、ギリギリありえない値とする
left, right = -1, (sum(C)+B)//N+1
while right-left > 2:
    left_mid, right_mid = (2*left+right)//3, (left+2*right)//3
    if check(left_mid) <= check(right_mid):
        # 右midの方が大きいから右を右midに変える
        right = right_mid
    else:
        left = left_mid

#print(left_mid, right_mid)

ans = min(check(left_mid), check(right_mid))
# leftとrightの間も入れる必要あるのか?　ない
print(ans)
0