結果

問題 No.2290 UnUnion Find
ユーザー t98slider
提出日時 2023-03-16 01:26:56
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,392 ms / 2,000 ms
コード長 1,121 bytes
コンパイル時間 267 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 13,952 KB
最終ジャッジ日時 2024-09-22 23:40:12
合計ジャッジ時間 61,908 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

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[rx] += self.parent_or_size[ry]
        self.parent_or_size[ry] = rx
        return True

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

uf = UnionFind(N)

l, r = 0, N - 1

for _ in range(Q):
    query = list(map(int, input().split()))
    if query[0] == 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 l == r:
            print(-1)
        elif uf.same(l, query[1] - 1):
            print(r + 1)
        else:
            print(l + 1)
0