結果

問題 No.1054 Union add query
ユーザー nephrologistnephrologist
提出日時 2020-05-16 09:43:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,211 ms / 2,000 ms
コード長 1,297 bytes
コンパイル時間 141 ms
コンパイル使用メモリ 81,308 KB
実行使用メモリ 141,732 KB
最終ジャッジ日時 2023-10-21 22:36:28
合計ジャッジ時間 7,976 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge9
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 33 ms
53,360 KB
testcase_01 AC 34 ms
53,360 KB
testcase_02 AC 38 ms
53,360 KB
testcase_03 AC 1,211 ms
101,688 KB
testcase_04 AC 1,196 ms
141,732 KB
testcase_05 AC 939 ms
89,624 KB
testcase_06 AC 674 ms
114,212 KB
testcase_07 AC 622 ms
114,212 KB
testcase_08 AC 585 ms
114,108 KB
testcase_09 AC 825 ms
129,528 KB
testcase_10 AC 318 ms
129,048 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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