結果

問題 No.1098 LCAs
コンテスト
ユーザー 👑 Kazun
提出日時 2020-06-26 23:40:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,048 ms / 2,000 ms
コード長 1,056 bytes
コンパイル時間 232 ms
コンパイル使用メモリ 82,268 KB
実行使用メモリ 187,736 KB
最終ジャッジ日時 2024-07-05 00:32:05
合計ジャッジ時間 14,203 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

from functools import lru_cache
from collections import deque

N=int(input())
E={i:[] for i in range(1,N+1)}
for _ in range(N-1):
    u,v=map(int,input().split())
    E[u].append(v)
    E[v].append(u)

T=[None]*(N+1)
U=[False]*(N+1)
D=[0]*(N+1)
F=[[] for _ in range(N+1)]

M={i:set() for i in range(N)}
Q=deque()
T[1]=0
M[0]={1}
Q.append(1)
while Q:
    v=Q.popleft()
    G=True
    for u in E[v]:
        if T[u]==None:
            Q.append(u)
            F[v].append(u)
            T[u]=1
            d=D[u]=D[v]+1
            M[d].add(u)
            G=False
    if G:
        U[v]=True

def h(L):
    S=0
    for p in L:
        S+=p*p

    T=sum(L+[1])
    return T*T-S

B=[0]*(N+1)
for d in range(N-1,-1,-1):
    for u in M[d]:
        if F[u]:
            C=1
            for a in F[u]:
                C+=B[a]
            B[u]=C
        else:
            B[u]=1

Z=[0]*(N+1)
for d in range(N-1,-1,-1):
    for u in M[d]:
        C=[]
        for v in F[u]:
            C.append(B[v])
        Z[u]=h(C)      


for i in range(1,N+1):
    print(Z[i])
0