結果

問題 No.938 賢人を探せ
ユーザー 0Kate0Kate
提出日時 2019-12-07 00:26:49
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 909 bytes
コンパイル時間 79 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 14,080 KB
最終ジャッジ日時 2024-06-06 12:53:36
合計ジャッジ時間 2,573 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
権限があれば一括ダウンロードができます

ソースコード

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