結果

問題 No.198 キャンディー・ボックス２
ユーザー FromBooskaFromBooska
提出日時 2023-10-17 13:13:44
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 41 ms / 1,000 ms
コード長 1,583 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 82,360 KB
実行使用メモリ 53,840 KB
最終ジャッジ日時 2024-09-17 11:16:30
合計ジャッジ時間 2,348 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,480 KB
testcase_01 AC 39 ms
52,688 KB
testcase_02 AC 39 ms
53,424 KB
testcase_03 AC 39 ms
53,248 KB
testcase_04 AC 39 ms
52,216 KB
testcase_05 AC 39 ms
52,708 KB
testcase_06 AC 38 ms
53,292 KB
testcase_07 AC 41 ms
52,744 KB
testcase_08 AC 40 ms
52,516 KB
testcase_09 AC 38 ms
52,124 KB
testcase_10 AC 38 ms
53,356 KB
testcase_11 AC 38 ms
53,204 KB
testcase_12 AC 38 ms
53,840 KB
testcase_13 AC 39 ms
52,408 KB
testcase_14 AC 38 ms
52,544 KB
testcase_15 AC 38 ms
53,500 KB
testcase_16 AC 39 ms
52,392 KB
testcase_17 AC 38 ms
52,344 KB
testcase_18 AC 39 ms
52,468 KB
testcase_19 AC 39 ms
53,500 KB
testcase_20 AC 39 ms
52,272 KB
testcase_21 AC 39 ms
53,692 KB
testcase_22 AC 39 ms
52,480 KB
testcase_23 AC 38 ms
53,092 KB
testcase_24 AC 39 ms
52,948 KB
testcase_25 AC 39 ms
52,660 KB
testcase_26 AC 39 ms
52,736 KB
testcase_27 AC 39 ms
52,364 KB
testcase_28 AC 39 ms
52,884 KB
testcase_29 AC 39 ms
53,344 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 二分探索に見えるが、単調ではなく2次曲線となるので三分探索
# おっとRE、入力例2、最初から下限と上限が2より近い場合だ
# 対処法は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)
    return count
    
# 谷、つまり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

if right-left <= 2:
    ans = 10**20
    for i in range(left, right+1):
        ans = min(ans, check(i))
    print(ans)
else:
    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