結果

問題 No.1098 LCAs
ユーザー brthyyjp
提出日時 2020-12-26 15:33:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 431 ms / 2,000 ms
コード長 734 bytes
コンパイル時間 472 ms
コンパイル使用メモリ 82,504 KB
実行使用メモリ 141,812 KB
最終ジャッジ日時 2024-09-24 18:35:59
合計ジャッジ時間 7,519 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
import io, os
input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline

n = int(input())
g = [[] for _ in range(n)]
for i in range(n-1):
    v, w = map(int, input().split())
    v, w = v-1, w-1
    g[v].append(w)
    g[w].append(v)

s = []
order = []
parent = [-1]*n
s.append(0)
while s:
    v = s.pop()
    order.append(v)
    for u in g[v]:
        if parent[v] != u:
            s.append(u)
            parent[u] = v
order.reverse()
C = [1]*n
for v in order:
    if parent[v] != -1:
        C[parent[v]] += C[v]
#print(C)
ans = [0]*n
for v in order:
    for u in g[v]:
        if u == parent[v]:
            continue
        ans[v] += 2*C[u]
        ans[v] += (C[v]-1-C[u])*C[u]
    ans[v] += 1
print(*ans, sep='\n')
0