結果

問題 No.1441 MErGe
ユーザー mkawa2
提出日時 2022-01-24 15:32:32
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 584 ms / 1,000 ms
コード長 1,009 bytes
コンパイル時間 417 ms
コンパイル使用メモリ 82,100 KB
実行使用メモリ 115,584 KB
最終ジャッジ日時 2024-12-14 11:52:17
合計ジャッジ時間 15,002 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    import sys

    def add(i, x):
        i += 1
        while i < n+1:
            table[i] += x
            i += i & -i

    def rank(x):
        if x == 0: return 0
        idx = 0
        d = 1 << n.bit_length()-1
        while d:
            if idx+d <= n and table[idx+d] < x:
                x -= table[idx+d]
                idx |= d
            d >>= 1
        return idx

    n, q = list(map(int, sys.stdin.readline().split()))
    aa = list(map(int, sys.stdin.readline().split()))

    cs = [0]
    for a in aa: cs.append(cs[-1]+a)

    table = [0]*(n+1)
    for i in range(1, n): add(i, 1)

    rr = list(range(1, n+1))

    for _ in range(q):
        t, l, r = list(map(int, sys.stdin.readline().split()))
        l -= 1
        if t == 1:
            i = rank(l)
            for _ in range(r-l-1):
                j = rr[i]
                add(j, -1)
                rr[i] = rr[j]
        else:
            i = rank(l)
            j = rank(r)
            print(cs[j]-cs[i])

main()
0