結果

問題 No.3780 Erase and Mex
コンテスト
ユーザー detteiuu
提出日時 2026-10-30 22:00:37
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
実行時間 -
コード長 1,448 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 87 ms
コンパイル使用メモリ 63,968 KB
実行使用メモリ 96,796 KB
最終ジャッジ日時 2026-10-30 22:00:49
合計ジャッジ時間 12,002 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 2 WA * 41
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from sys import stdin
input = stdin.readline

def func(n, C, cnt):
    for i in reversed(range(n, N*2+10)):
        cnt += C[i]
        C[i] = 0
    right = 0
    for i in range(n):
        if right == i:
            right += 1
        if 1 <= C[i]:
            rem = C[i]-1
            C[i] = 1
            if 1 <= i:
                cnt += rem*(1<<(i-1))
            else:
                cnt += rem
        else:
            if 1 <= i:
                cost = 1<<(i-1)
            else:
                cost = 1
            if cost <= cnt:
                cnt -= cost
                C[i] = 1
            else:
                cost -= cnt
                cnt = 0
                while 1 <= cost and right < N*2+10:
                    if 1 <= C[right]:
                        cost -= 1
                        C[right] -= 1
                    if C[right] == 0:
                        right += 1
                if cost == 0:
                    C[i] = 1
                else:
                    return False
    return True

for _ in range(int(input())):
    N = int(input())
    A = list(map(int, input().split()))

    C = [0]*(N*2+10)
    cnt = 0
    for a in A:
        if a < N*2+10:
            C[a] += 1
        else:
            cnt += 1

    left = 0
    right = N*3
    while left+1 < right:
        mid = (left+right)//2
        if func(mid, C[:], cnt):
            left = mid
        else:
            right = mid

    print(left)
0