結果
| 問題 |
No.2427 Tree Distance Two
|
| コンテスト | |
| ユーザー |
Cecil
|
| 提出日時 | 2023-08-18 22:31:04 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 695 bytes |
| コンパイル時間 | 232 ms |
| コンパイル使用メモリ | 82,048 KB |
| 実行使用メモリ | 369,152 KB |
| 最終ジャッジ日時 | 2024-11-28 08:27:18 |
| 合計ジャッジ時間 | 63,801 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| other | AC * 15 TLE * 20 |
ソースコード
from collections import deque
from collections import defaultdict as dd
n = int(input())
G = [ [] for _ in range(n+1) ]
for _ in range(n-1):
u,v = map(int, input().split())
G[u].append(v)
G[v].append(u)
def bfs(start):
dist = [-1] * (n+1)
dist[start] = 0
que = deque()
que.append(start)
cnt = 0
while que:
v = que.popleft()
for v2 in G[v]:
if dist[v2] != -1:
continue
dist[v2] = dist[v] + 1
if dist[v2] == 2:
cnt += 1
if dist[v2] > 2:
return cnt
que.append(v2)
return cnt
for i in range(1, n+1):
ans = bfs(i)
print(ans)
Cecil