結果

問題 No.2290 UnUnion Find
ユーザー Yukino DX.Yukino DX.
提出日時 2024-09-13 22:17:12
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,271 bytes
コンパイル時間 383 ms
コンパイル使用メモリ 81,972 KB
実行使用メモリ 85,120 KB
最終ジャッジ日時 2024-09-13 22:17:22
合計ジャッジ時間 6,221 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 42 ms
59,520 KB
testcase_01 AC 44 ms
54,272 KB
testcase_02 AC 519 ms
77,696 KB
testcase_03 AC 422 ms
79,616 KB
testcase_04 TLE -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import random


class UnionFindTree:
    def __init__(self, n):
        self.par = [-1] * n
        self.rank = [0] * n
        self.sz = [1] * n

    def root(self, v):
        if self.par[v] == -1:
            return v
        self.par[v] = self.root(self.par[v])
        return self.par[v]

    def same(self, v1, v2):
        return self.root(v1) == self.root(v2)

    def size(self, v):
        return self.sz[self.root(v)]

    def unite(self, v1, v2):
        root1, root2 = self.root(v1), self.root(v2)
        if root1 == root2:
            return
        if self.rank[root1] < self.rank[root2]:
            root1, root2 = root2, root1
        self.par[root2] = root1
        if self.rank[root1] == self.rank[root2]:
            self.rank[root1] += 1
        self.sz[root1] += self.sz[root2]


n, q = map(int, input().split())
uf = UnionFindTree(n)
for i in range(q):
    query = list(map(int, input().split()))
    if query[0] == 1:
        u, v = query[1] - 1, query[2] - 1
        uf.unite(u, v)
    else:
        v = query[1] - 1
        if uf.size(v) == n:
            print(-1)
        else:
            while True:
                u = random.randrange(0, n)
                if not uf.same(u, v):
                    print(u + 1)
                    break
0