提出 #28347336


ソースコード 拡げる

from collections import Counter
import sys
input = sys.stdin.readline

N, K, L = map(int, input().split())


class UnionFind:
    def __init__(self, N):
        self.par = [-1] * (N + 1)

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

    def unite(self, x, y):
        x = self.find(x)
        y = self.find(y)

        if x == y:
            return False
        if x > y:
            x, y = y, x

        self.par[x] += self.par[y]
        self.par[y] = x
        return True

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

    def members(self, x):
        p = self.find(x)
        return [i for i in range(1, N + 1) if self.find(i) == p]

    def size(self, x):
        return -self.par[self.find(x)]


uf1 = UnionFind(N)
uf2 = UnionFind(N)
for _ in range(K):
    p, q = map(int, input().split())
    uf1.unite(p, q)

ans = [1 for _ in range(N + 1)]
for _ in range(L):
    r, s = map(int, input().split())
    uf2.unite(r, s)

pars = [(uf1.find(i), uf2.find(i)) for i in range(1, N + 1)]
C = Counter(pars)
print(*[C[p] for p in pars])

提出情報

提出日時
問題 D - 連結
ユーザ terasa
言語 PyPy3 (7.3.0)
得点 400
コード長 1240 Byte
結果 AC
実行時間 404 ms
メモリ 164248 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 400 / 400
結果
AC × 3
AC × 18
セット名 テストケース
Sample subtask0_0.txt, subtask0_1.txt, subtask0_2.txt
All subtask0_0.txt, subtask0_1.txt, subtask0_2.txt, subtask1_0.txt, subtask1_1.txt, subtask1_10.txt, subtask1_11.txt, subtask1_12.txt, subtask1_13.txt, subtask1_14.txt, subtask1_2.txt, subtask1_3.txt, subtask1_4.txt, subtask1_5.txt, subtask1_6.txt, subtask1_7.txt, subtask1_8.txt, subtask1_9.txt
ケース名 結果 実行時間 メモリ
subtask0_0.txt AC 98 ms 64348 KiB
subtask0_1.txt AC 57 ms 64924 KiB
subtask0_2.txt AC 58 ms 64052 KiB
subtask1_0.txt AC 195 ms 76584 KiB
subtask1_1.txt AC 337 ms 164248 KiB
subtask1_10.txt AC 188 ms 76496 KiB
subtask1_11.txt AC 314 ms 141436 KiB
subtask1_12.txt AC 299 ms 124988 KiB
subtask1_13.txt AC 322 ms 149588 KiB
subtask1_14.txt AC 404 ms 136332 KiB
subtask1_2.txt AC 271 ms 131080 KiB
subtask1_3.txt AC 313 ms 152176 KiB
subtask1_4.txt AC 393 ms 132248 KiB
subtask1_5.txt AC 200 ms 75632 KiB
subtask1_6.txt AC 291 ms 131192 KiB
subtask1_7.txt AC 312 ms 154992 KiB
subtask1_8.txt AC 317 ms 160816 KiB
subtask1_9.txt AC 390 ms 117496 KiB