結果

問題 No.556 仁義なきサルたち
コンテスト
ユーザー Pump0129
提出日時 2018-05-24 00:34:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,296 bytes
コンパイル時間 382 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 16,000 KB
最終ジャッジ日時 2024-06-28 16:37:22
合計ジャッジ時間 10,448 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17 TLE * 2 -- * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

def get_root(monkey_data, monkey_id):
    for monkey_root in monkey_data:
        if monkey_id in monkey_data[monkey_root]:
            return monkey_root


if __name__ == "__main__":
    n, m = map(int, input().split(" "))
    root_data = {i + 1: (i + 1,) for i in range(n)}
    for i in range(m):
        a, b = map(int, input().split(" "))
        a_root = get_root(root_data, a)
        b_root = get_root(root_data, b)
        if a_root != b_root:
            if len(root_data[a_root]) != len(root_data[b_root]):
                if len(root_data[a_root]) < len(root_data[b_root]):
                    root_data.update({b_root: root_data[b_root] + root_data.pop(a_root)})
                else:
                    root_data.update({a_root: root_data[a_root] + root_data.pop(b_root)})
            else:
                if a_root < b_root:
                    root_data.update({a_root: root_data[a_root] + root_data.pop(b_root)})
                else:
                    root_data.update({b_root: root_data[b_root] + root_data.pop(a_root)})
    monkey_root_list = [0 for i in range(n)]
    for monkey_key in root_data:
        for monkey_num in root_data[monkey_key]:
            monkey_root_list[monkey_num - 1] = monkey_key
    print(*monkey_root_list, sep="\n")
0