結果

問題 No.1054 Union add query
コンテスト
ユーザー nephrologist
提出日時 2020-05-16 09:44:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 779 ms / 2,000 ms
コード長 1,343 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 143,488 KB
最終ジャッジ日時 2024-09-22 00:14:44
合計ジャッジ時間 5,149 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input = sys.stdin.buffer.readline
n, q = map(int, input().split())

# union-findを魔改造
groups = [[i] for i in range(n + 1)]  # groupsには同グループすべての頂点を入れる

par = [-1] * (n + 1)  # parには根か、-sizeが入る
add_edge = [0] * (n + 1)
add_root = [0] * (n + 1)


def find(x):
    if par[x] < 0:
        return x
    else:
        y = par[x]
        par[x] = find(y)
        return par[x]


def size(x):
    return -par[find(x)]


def unite(x, y):
    if find(x) == find(y):
        return False
    X = find(x)
    Y = find(y)
    if size(x) < size(y):  # 必ずyのsizeが小さくする
        x, y = y, x
        X = find(x)
        Y = find(y)
    par[X] += par[Y]
    par[Y] = X
    # 各々の足し算結果の調整
    rx = add_root[X]
    ry = add_root[Y]
    for i in range(len(groups[Y])):
        e = groups[Y][i]
        oldpy = add_edge[e]
        newpy = oldpy + (ry - rx)
        add_edge[e] = newpy
    add_root[Y] = 0
    # group管理の調整
    groups[X].extend(groups[Y])
    groups[Y] = []
    return True


def query(x):
    X = find(x)
    return add_edge[x] + add_root[X]


for _ in range(q):
    t, a, b = map(int, input().split())
    if t == 1:
        unite(a, b)
    elif t == 2:
        X = find(a)
        add_root[X] += b
    else:
        print(query(a))
0