結果

問題 No.1441 MErGe
ユーザー lam6er
提出日時 2025-04-15 21:32:02
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,198 bytes
コンパイル時間 417 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 155,192 KB
最終ジャッジ日時 2025-04-15 21:34:26
合計ジャッジ時間 9,998 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 9 TLE * 3 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
import random

random.seed(0)

class Node:
    __slots__ = ['val', 'prio', 'left', 'right', 'size', 'sum']
    def __init__(self, val):
        self.val = val
        self.prio = random.randint(0, 1 << 60)
        self.left = None
        self.right = None
        self.size = 1
        self.sum = val

def get_size(node):
    return node.size if node else 0

def get_sum(node):
    return node.sum if node else 0

def update(node):
    if node:
        node.size = 1 + get_size(node.left) + get_size(node.right)
        node.sum = node.val + get_sum(node.left) + get_sum(node.right)

def split(node, k):
    if not node:
        return (None, None)
    left_size = get_size(node.left)
    if left_size >= k:
        l, r = split(node.left, k)
        node.left = r
        update(node)
        return (l, node)
    else:
        l, r = split(node.right, k - left_size - 1)
        node.right = l
        update(node)
        return (node, r)

def merge(a, b):
    if not a:
        return b
    if not b:
        return a
    if a.prio > b.prio:
        a.right = merge(a.right, b)
        update(a)
        return a
    else:
        b.left = merge(a, b.left)
        update(b)
        return b

def main():
    data = sys.stdin.read().split()
    ptr = 0
    n = int(data[ptr])
    ptr += 1
    q = int(data[ptr])
    ptr += 1
    A = list(map(int, data[ptr:ptr + n]))
    ptr += n

    root = None
    for a in A:
        new_node = Node(a)
        root = merge(root, new_node)

    outputs = []
    for _ in range(q):
        T = int(data[ptr])
        ptr += 1
        l = int(data[ptr])
        ptr += 1
        r = int(data[ptr])
        ptr += 1
        if T == 1:
            left_part, right_part = split(root, r)
            left, mid = split(left_part, l - 1)
            sum_mid = get_sum(mid)
            new_node = Node(sum_mid)
            root = merge(merge(left, new_node), right_part)
        else:
            left_part, right_part = split(root, r)
            left, mid = split(left_part, l - 1)
            outputs.append(str(get_sum(mid)))
            root = merge(merge(left, mid), right_part)
    print('\n'.join(outputs))

if __name__ == "__main__":
    main()
0