結果

問題 No.938 賢人を探せ
コンテスト
ユーザー 0Kate
提出日時 2019-12-07 00:26:49
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 909 bytes
コンパイル時間 97 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 14,204 KB
最終ジャッジ日時 2024-12-24 04:41:41
合計ジャッジ時間 3,153 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UnionFind:
    def __init__(self):
        self.parent = {}

    def root(self, name):
        if self.parent[name] == name:
            return name
        else:
            self.parent[name] = self.parent[self.parent[name]] 
            return self.root(self.parent[name])

    def unite(self, a, b):
        if a not in self.parent:
            self.new_node(a)

        if b not in self.parent:
            self.new_node(b)

        a_root = self.root(a)
        b_root = self.root(b)
        self.parent[a_root] = b_root

    def new_node(self, node):
        self.parent[node] = node

    def roots(self):
        return [name for name, parent in self.parent.items() if parent == name]
            

if __name__ == '__main__':
    union_find = UnionFind()

    n = int(input())
    for i in range(n):
        a, b = input().split(' ')
        union_find.unite(a, b)

    print(union_find.roots())
0