結果

問題 No.3309 Aging Railway
コンテスト
ユーザー i_taku
提出日時 2026-01-21 13:49:34
言語 PyPy3
(7.3.17)
結果
AC  
実行時間 2,056 ms / 3,000 ms
コード長 1,297 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 241 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 261,004 KB
最終ジャッジ日時 2026-01-21 13:50:08
合計ジャッジ時間 27,159 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

class UnionFind:
    def __init__(self, n):
        self._n = n
        self._parent = [-1] * n

    def _find(self, x):
        if self._parent[x] < 0:
            return x
        self._parent[x] = self._find(self._parent[x])
        return self._parent[x]

    def union(self, x, y):
        x, y = self._find(x), self._find(y)
        if x == y:
            return
        if self._parent[y] < self._parent[x]:
            x, y = y, x
        self._parent[x] += self._parent[y]
        self._parent[y] = x

    def same(self, x, y):
        return self._find(x) == self._find(y)

N, M = map(int, input().split())
E = [tuple(map(lambda x: int(x - 1), map(int, input().split()))) for _ in range(N - 1)]
ST = [tuple(map(lambda x: int(x - 1), map(int, input().split()))) for _ in range(M)]
ok = [0] * M
ng = [N] * M
for _ in range(15):
    idx = [[] for _ in range(N + 1)]
    for i in range(M):
        idx[(ok[i] + ng[i]) // 2].append(i)

    uf = UnionFind(N)    
    for mid in range(N - 2, -1, -1):
        uf.union(*E[mid])
        for i in idx[mid]:
            s, t = ST[i]
            if uf.same(s, t):
                ok[i] = mid
            else:
                ng[i] = mid

ok.sort(reverse=True)
for i in range(N - 1):
    while ok and ok[-1] == i:
        ok.pop()
    print(len(ok))
0