結果

問題 No.1441 MErGe
ユーザー lam6er
提出日時 2025-04-15 21:31:40
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,333 bytes
コンパイル時間 342 ms
コンパイル使用メモリ 82,204 KB
実行使用メモリ 126,600 KB
最終ジャッジ日時 2025-04-15 21:34:01
合計ジャッジ時間 7,901 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 6 TLE * 3 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
import random
random.seed(42)

sys.setrecursionlimit(1 << 25)

class Node:
    __slots__ = ['val', 'sum', 'size', 'priority', 'left', 'right']
    def __init__(self, val):
        self.val = val
        self.sum = val
        self.size = 1
        self.priority = random.randint(1, 10**18)
        self.left = None
        self.right = None

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, key):
    if node is None:
        return (None, None)
    current_key = get_size(node.left) + 1
    if current_key <= key:
        left, right = split(node.right, key - current_key)
        node.right = left
        update(node)
        return (node, right)
    else:
        left, right = split(node.left, key)
        node.left = right
        update(node)
        return (left, node)

def merge(left, right):
    if left is None:
        return right
    if right is None:
        return left
    if left.priority > right.priority:
        left.right = merge(left.right, right)
        update(left)
        return left
    else:
        right.left = merge(left, right.left)
        update(right)
        return right

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    N = int(input[ptr])
    ptr +=1
    Q = int(input[ptr])
    ptr +=1
    A = list(map(int, input[ptr:ptr+N]))
    ptr +=N

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

    for _ in range(Q):
        T = int(input[ptr])
        ptr +=1
        l = int(input[ptr])
        ptr +=1
        r = int(input[ptr])
        ptr +=1

        if T == 1:
            left, temp = split(root, l-1)
            middle, right = split(temp, r - (l-1))
            sum_middle = middle.sum if middle else 0
            new_node = Node(sum_middle)
            root = merge(merge(left, new_node), right)
        else:
            left, temp = split(root, l-1)
            middle, right = split(temp, r - (l-1))
            print(middle.sum if middle else 0)
            root = merge(merge(left, middle), right)

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