結果

問題 No.1098 LCAs
ユーザー 👑 KazunKazun
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 45 ms
55,368 KB
testcase_01 AC 46 ms
56,200 KB
testcase_02 AC 46 ms
56,436 KB
testcase_03 AC 46 ms
55,712 KB
testcase_04 AC 46 ms
55,972 KB
testcase_05 AC 46 ms
57,176 KB
testcase_06 AC 46 ms
57,184 KB
testcase_07 AC 45 ms
56,044 KB
testcase_08 AC 45 ms
55,728 KB
testcase_09 AC 46 ms
55,796 KB
testcase_10 AC 46 ms
55,832 KB
testcase_11 AC 47 ms
57,120 KB
testcase_12 AC 45 ms
56,844 KB
testcase_13 AC 82 ms
74,972 KB
testcase_14 AC 85 ms
76,724 KB
testcase_15 AC 85 ms
77,140 KB
testcase_16 AC 79 ms
74,560 KB
testcase_17 AC 79 ms
74,468 KB
testcase_18 AC 955 ms
154,664 KB
testcase_19 AC 931 ms
154,700 KB
testcase_20 AC 911 ms
153,968 KB
testcase_21 AC 949 ms
153,856 KB
testcase_22 AC 936 ms
153,460 KB
testcase_23 AC 703 ms
179,952 KB
testcase_24 AC 696 ms
180,088 KB
testcase_25 AC 656 ms
181,552 KB
testcase_26 AC 643 ms
184,296 KB
testcase_27 AC 641 ms
181,160 KB
testcase_28 AC 1,047 ms
184,808 KB
testcase_29 AC 1,048 ms
187,736 KB
testcase_30 AC 1,038 ms
184,216 KB
権限があれば一括ダウンロードができます

ソースコード

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