結果

問題 No.2290 UnUnion Find
ユーザー t98slidert98slider
提出日時 2023-03-16 01:19:32
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 1,212 bytes
コンパイル時間 353 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 13,696 KB
最終ジャッジ日時 2024-09-22 23:38:11
合計ジャッジ時間 15,233 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 29 ms
10,624 KB
testcase_01 WA -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 AC 1,247 ms
12,672 KB
testcase_12 WA -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 AC 1,269 ms
12,160 KB
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
testcase_45 RE -
testcase_46 RE -
権限があれば一括ダウンロードができます

ソースコード

diff #

class UnionFind():
    def __init__(self, n):
        self.parent_or_size = [-1] * n

    def leader(self, x):
        if self.parent_or_size[x] < 0: return x
        else:
            self.parent_or_size[x] = self.leader(self.parent_or_size[x])
            return self.parent_or_size[x]

    def same(self, x, y):
        return self.leader(x) == self.leader(y)

    def merge(self, x, y):
        rx = self.leader(x)
        ry = self.leader(y)
        if rx == ry: return False
        if self.parent_or_size[rx] > self.parent_or_size[ry]:
            rx, ry = ry, rx
        self.parent_or_size[x] += self.parent_or_size[y]
        self.parent_or_size[y] = x
        return True

    def size(self, x):
        return -self.parent_or_size[self.leader(x)]

N, Q = map(int, input().split())

uf = UnionFind(N)

l, r = 0, N - 1

for _ in range(Q):
    query = list(map(int, input().split()))
    t = query[0]
    if t == 1:
        uf.merge(query[1] - 1, query[2] - 1)
        while l != uf.leader(l): l += 1
        while r != uf.leader(r): r -= 1
    else:
        if uf.size(0) == N:
            print(-1)
        elif uf.same(l, query[1] - 1):
            print(r + 1)
        else:
            print(l + 1)
0