結果

問題 No.1054 Union add query
ユーザー tktk_snsntktk_snsn
提出日時 2021-02-03 23:21:48
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 955 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,060 KB
実行使用メモリ 264,508 KB
最終ジャッジ日時 2024-06-30 06:39:07
合計ジャッジ時間 5,441 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other TLE * 1 -- * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)

N, Q = map(int, input().split())
query = tuple(tuple(map(int, input().split())) for _ in range(Q))

root = [-1] * N
member = [[i] for i in range(N)]
leader_val = [0] * N
each_val = [0] * N


def find(x):
    if root[x] < 0:
        return x
    root[x] = res = find(root[x])
    return res


def unite(x, y):
    x = find(x)
    y = find(y)
    if x == y:
        return
    # x:small, y:big
    if -root[x] > -root[y]:
        x, y = y, x

    xval = leader_val[x]
    yval = leader_val[y]
    leader_val[y] = 0

    root[x] += root[y]
    root[y] = x
    while member[y]:
        i = member[y].pop()
        each_val[i] += yval - xval
        member[x].append(i)


for t, a, b in query:
    if t == 1:
        unite(a - 1, b - 1)
    elif t == 2:
        x = find(a - 1)
        leader_val[x] += b
    else:
        x = find(a - 1)
        print(leader_val[x] + each_val[a-1])
0