結果

問題 No.277 根掘り葉掘り
ユーザー takakin
提出日時 2020-06-22 23:49:31
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 874 bytes
コンパイル時間 141 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 38,272 KB
最終ジャッジ日時 2024-07-03 19:01:50
合計ジャッジ時間 10,790 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 10 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input=lambda: sys.stdin.readline().rstrip()
n=int(input())
edge=[[] for i in range(n)]
for i in range(n-1):
  a,b=map(int,input().split())
  edge[a-1].append(b-1)
  edge[b-1].append(a-1)

inf=float("inf")
R=[-1]*n
R[0]=0
Par=[inf]*n
Par[0]=-1
Deg=[0]*n
Deg[0]=0
Chk=[0]

while Chk:
  c=Chk.pop()
  for next in edge[c]:
    if R[next]==-1:
      R[next]=R[c]+1
      Par[next]=c
      Deg[next]+=1
      Chk.append(next)

from collections import deque
TS=list(v for v in range(n) if Deg[v]==0)
D=deque(TS)
while D:
  v=D.popleft()
  for t in edge[v]:
    if t!=Par[v]:
      Deg[t]-=1
      if Deg[t]==0:
        D.append(t)
        TS.append(t)

Used=[False]*n
C=[inf]*n
for i in reversed(range(n)):
  v=TS[i]
  Used[v]=True
  if C[v]==inf:
    C[v]=0
  for g in edge[v]:
    if not Used[g]:
      C[g]=min(C[g],C[v]+1)

for i in range(n):
  print(min(R[i],C[i]))
0