結果

問題 No.277 根掘り葉掘り
ユーザー roarisroaris
提出日時 2019-12-07 22:42:47
言語 Python3
(3.10.1 + numpy 1.22.3 + scipy 1.8.0)
結果
AC  
実行時間 748 ms / 3,000 ms
コード長 1,254 bytes
コンパイル時間 65 ms
使用メモリ 134,888 KB
最終ジャッジ日時 2023-01-15 00:55:11
合計ジャッジ時間 9,211 ms
ジャッジサーバーID
(参考情報)
judge13 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 18 ms
8,172 KB
testcase_01 AC 18 ms
8,016 KB
testcase_02 AC 18 ms
8,016 KB
testcase_03 AC 19 ms
8,268 KB
testcase_04 AC 18 ms
8,264 KB
testcase_05 AC 18 ms
8,132 KB
testcase_06 AC 19 ms
8,272 KB
testcase_07 AC 19 ms
8,264 KB
testcase_08 AC 18 ms
8,112 KB
testcase_09 AC 748 ms
134,888 KB
testcase_10 AC 498 ms
25,216 KB
testcase_11 AC 574 ms
27,076 KB
testcase_12 AC 685 ms
81,716 KB
testcase_13 AC 661 ms
27,224 KB
testcase_14 AC 578 ms
31,264 KB
testcase_15 AC 612 ms
39,716 KB
testcase_16 AC 610 ms
33,820 KB
testcase_17 AC 598 ms
33,084 KB
testcase_18 AC 599 ms
32,996 KB
testcase_19 AC 586 ms
31,492 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**6)
from collections import deque

def bfs1():
    q = deque([0])
    dist = [-1]*N
    dist[0] = 0
    
    while q:
        v = q.popleft()
        
        for nv in adj_list[v]:
            if dist[nv]==-1:
                dist[nv] = dist[v]+1
                q.append(nv)
    
    return dist
    
def dfs(v, p):
    is_leaf = True
    res = 10**18
    
    for nv in adj_list[v]:
        if nv==p:
            continue
        
        is_leaf = False
        res  = min(res, dfs(nv, v)+1)
    
    if is_leaf:
        res = 0
    
    from_leaf[v] = res
    
    return res

def bfs2():
    q = deque([0])
    visited = [-1]*N
    visited[0] = 1
    
    while q:
        v = q.popleft()
        
        for nv in adj_list[v]:
            if visited[nv]==-1:
                visited[nv] = 1
                from_leaf[nv] = min(from_leaf[nv], from_leaf[v]+1)
                q.append(nv)

N = int(input())
adj_list = [[] for _ in range(N)]

for _ in range(N-1):
    xi, yi = map(int, input().split())
    adj_list[xi-1].append(yi-1)
    adj_list[yi-1].append(xi-1)

from_root = bfs1()
from_leaf = [-1]*N
dfs(0, -1)
bfs2()

for i in range(N):
    print(min(from_root[i], from_leaf[i]))
0