結果

問題 No.2935 Division Into 3 (Mex Edition)
コンテスト
ユーザー amesyu
提出日時 2024-10-12 01:30:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 3,035 ms / 5,000 ms
コード長 1,960 bytes
コンパイル時間 219 ms
コンパイル使用メモリ 82,256 KB
実行使用メモリ 292,612 KB
最終ジャッジ日時 2024-10-12 01:31:32
合計ジャッジ時間 58,553 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import bisect_left, bisect_right

LOG = 17
MAX_N = 2**LOG # 131072
INF = 10**9

N = int(input())
A = [*map(int, input().split())]
ap = [[] for _ in range(MAX_N)]
for i in range(N):
    ap[A[i]].append(i)

treeL = [[[] for _ in range(2*MAX_N)] for __ in range(3)]
treeR = [[[] for _ in range(2*MAX_N)] for __ in range(3)]
interval = [(-1, -1)] * 2 * MAX_N

for t in range(1, 2*MAX_N):
    if t == 1: l, r = 0, MAX_N
    else:
        pl, pr = interval[t//2]
        if t&1: l, r = (pl + pr) >> 1, pr
        else: l, r = pl, (pl + pr) >> 1
        
    idx = []
    for x in range(l, r):
        for i in ap[x]:
            idx.append(i)
    idx.sort()
    
    for c in range(1, 4):
        cnt = [0] * (r - l)
        R, now = 0, 0
        for i in range(len(idx)):
            assert i <= R

            while R < len(idx) and now < r - l:
                pos = A[idx[R]] - l
                cnt[pos] += 1
                if cnt[pos] == c:
                    now += 1
                R += 1

            if now == r - l:
                treeL[c-1][t].append(idx[i])
                treeR[c-1][t].append(idx[R-1])
            
            if cnt[A[idx[i]]-l] == c:
                now -= 1

            cnt[A[idx[i]]-l] -= 1

        treeL[c-1][t].append(INF)
        treeR[c-1][t].append(INF)
    
    interval[t] = (l, r)

def query(l, r):
    mex = []
    for c in range(1, 4):
        ptr = 2
        while ptr < 2*MAX_N:
            L, R = interval[ptr]
            p = bisect_left(treeL[c-1][ptr], l)
            p = treeR[c-1][ptr][p]
            if p < r: ptr = (ptr^1)*2
            else: ptr <<= 1
        mex.append(ptr//2 - MAX_N)
    return mex

Q = int(input())
ansi = 0
for _ in range(Q):
    alpha, beta = map(int, input().split())
    l, r = alpha ^ ansi, beta ^ ansi
    mex1, mex2, mex3 = query(l-1, r)
    ansi = mex1+mex2+mex3
    if mex2 == 0: ansi = min(ansi, r-l-1)
    if mex3 == 0: ansi = min(ansi, r-l)
    print(ansi)
0