結果
問題 | No.277 根掘り葉掘り |
ユーザー |
|
提出日時 | 2020-06-09 09:03:27 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 977 ms / 3,000 ms |
コード長 | 884 bytes |
コンパイル時間 | 261 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 36,404 KB |
最終ジャッジ日時 | 2024-12-30 18:40:55 |
合計ジャッジ時間 | 13,609 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
from collections import dequeN = int(input())int1 = lambda x: int(x) - 1T = [[] for _ in range(N)]for _ in range(N - 1):a, b = map(int1, input().split())T[a].append(b)T[b].append(a)# 根からの距離(dfs)d = deque([0])root_distance = [None] * Nroot_distance[0] = 0while d:v = d.pop()for x in T[v]:if root_distance[x] is None:root_distance[x] = root_distance[v] + 1d.append(x)# print(root_distance)# 葉からの距離(bfs)leaves = {i for i in range(N) if len(T[i]) == 1}leaf_distance = [0 if i in leaves else None for i in range(N)]d = deque(leaves)while d:v = d.popleft()for x in T[v]:if leaf_distance[x] is None:leaf_distance[x] = leaf_distance[v] + 1d.append(x)# print(leaf_distance)for i in range(N):print(min(root_distance[i], leaf_distance[i]))