結果

問題 No.2427 Tree Distance Two
ユーザー timi
提出日時 2023-08-18 21:39:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 857 ms / 2,000 ms
コード長 658 bytes
コンパイル時間 360 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 184,240 KB
最終ジャッジ日時 2024-11-28 06:10:26
合計ジャッジ時間 14,827 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

##N,K=map(int, input().split())
#A=list(map(int, input().split()))
N=int(input())
D=[[]for _ in range(N)]
C=[[]for _ in range(N)]
for i in range(N-1):
  u,v=map(int, input().split())
  u-=1;v-=1
  D[u].append(v);D[v].append(u)
from collections import deque
d=deque()
d.append(0)
E=[-1]*N;P=[-1]*N
E[0]=0 
while d:
  now=d.popleft()
  for nex in D[now]:
    if E[nex]==-1:
      d.append(nex)
      E[nex]=E[now]+1 
      P[nex]=now
      C[now].append(nex)
CC=[]
for i in range(N):
  CC.append(len(C[i]))
  
for i in range(N):
  ans=0
  if E[i]>1:
    ans+=1 
  if E[i]>0:
    p=P[i]
    ans+=len(C[p])-1 
  CD=C[i]
  for c in CD:
    ans+=CC[c]
  print(ans)
0