結果

問題 No.1054 Union add query
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2022-04-06 23:28:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 936 ms / 2,000 ms
コード長 1,481 bytes
コンパイル時間 197 ms
コンパイル使用メモリ 82,268 KB
実行使用メモリ 189,044 KB
最終ジャッジ日時 2024-05-05 20:12:19
合計ジャッジ時間 8,655 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
52,352 KB
testcase_01 AC 38 ms
52,736 KB
testcase_02 AC 37 ms
52,608 KB
testcase_03 AC 866 ms
150,260 KB
testcase_04 AC 936 ms
189,044 KB
testcase_05 AC 732 ms
140,788 KB
testcase_06 AC 598 ms
160,192 KB
testcase_07 AC 601 ms
160,548 KB
testcase_08 AC 557 ms
160,352 KB
testcase_09 AC 588 ms
177,796 KB
testcase_10 AC 524 ms
176,044 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Unionfind:
     
    def __init__(self,n):
        self.uf = [-1]*n
        self.p = [[i] for i in range(n)]
        self.count = [0]*n
    def find(self,x):
        if self.uf[x] < 0:
            return x
        else:
            self.uf[x] = self.find(self.uf[x])
            return self.uf[x]
 
    def same(self,x,y):
        return self.find(x) == self.find(y)
 
    def union(self,x,y):
        x = self.find(x)
        y = self.find(y)
        if x == y:
            return False
        if self.uf[x] > self.uf[y]:
            x,y = y,x

        cx = self.count[x]
        cy = self.count[y]
        for i in self.p[y]:
            if i != y:
                self.count[i] += cy-cx
            else:
                self.count[i] += -cx
            self.p[x].append(i)
        self.p[y] = []
        self.uf[x] += self.uf[y]
        self.uf[y] = x
        return True
 
    def size(self,x):
        x = self.find(x)
        return -self.uf[x]

    def add(self,a,b):
        self.count[self.find(a)] += b

    def solve(self,a):
        num = self.count[a]
        if self.find(a) != a:
            num += self.count[self.find(a)]
        return num


n,q = map(int,input().split())


uf = Unionfind(n)

Q = [list(map(int,input().split())) for i in range(q)]
for t,a,b in Q:

    if t == 1:
        a,b = a-1,b-1
        if uf.same(a,b):
            continue

        uf.union(a,b)

    elif t == 2:
        uf.add(a-1,b)
    
    else:
        print(uf.solve(a-1))
0