結果

問題 No.1639 最小通信路
コンテスト
ユーザー sash0
提出日時 2021-08-22 23:26:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 100 ms / 2,000 ms
コード長 1,340 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 77,312 KB
最終ジャッジ日時 2024-11-06 23:02:51
合計ジャッジ時間 5,001 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UnionFind:
    def __init__(self, n):
        self._par = list(i for i in range(n))
        self._rank = list(0 for _ in range(n))
        self._count = list(1 for _ in range(n))

    def parent(self, a):
        if self._par[a] == a:
            return a
        else:
            self._par[a] = self.parent(self._par[a])
            return self._par[a]

    def size(self, a):
        return self._count[self.parent(a)]

    def merge(self, a, b):
        a = self.parent(a)
        b = self.parent(b)

        if a == b:
            return

        if self._rank[a] > self._rank[b]:
            self._par[b] = a
            self._count[a] += self._count[b]
        else:
            self._par[a] = b
            self._count[b] += self._count[a]
            if self._rank[a] == self._rank[b]:
                self._rank[b] += 1

    def same(self, a, b):
        return self.parent(a) == self.parent(b)


if __name__ == '__main__':
    n = int(input().strip())
    li = []
    for i in range(n*(n-1)//2):
        a, b, c = map(int, input().strip().split())
        a -= 1
        b -= 1
        li.append((c, a, b))

    li.sort()

    uf = UnionFind(n)

    ans = 0
    for (c, a, b) in li:
        if uf.size(0) == n:
            break

        if not uf.same(a, b):
            uf.merge(a, b)
            ans = c

    print(ans)
0