結果

問題 No.556 仁義なきサルたち
ユーザー NieR
提出日時 2018-06-18 15:57:00
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 77 ms / 2,000 ms
コード長 990 bytes
コンパイル時間 85 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 11,264 KB
最終ジャッジ日時 2024-06-30 16:56:08
合計ジャッジ時間 2,219 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

class union_find:
    def __init__(self,size):
        self.table = [-1 for i in range(size)]

    def find(self,n):
        while self.table[n] >= 0:
            n = self.table[n]
        return n

    def merge(self,x,y):
        x = self.find(x)
        y = self.find(y)
        if self.table[x] < self.table[y]:
            self.table[x] += self.table[y]
            self.table[y] = x
        elif self.table[x] == self.table[y]:
            if x > y:
                self.table[y] += self.table[x]
                self.table[x] = y
            elif x == y:
                pass
            else:
                self.table[x] += self.table[y]
                self.table[y] = x
        else:
            self.table[y] += self.table[x]
            self.table[x] = y


A, B = map(int, input().split())
aggregation = union_find(A)
for i in range(B):
    C,D = map(int, input().split())
    C = C-1
    D = D-1
    aggregation.merge(C,D)
for u in range(A):
    print(aggregation.find(u)+1)
0