結果
問題 |
No.2638 Initial fare
|
ユーザー |
|
提出日時 | 2024-02-20 00:38:44 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 672 bytes |
コンパイル時間 | 151 ms |
コンパイル使用メモリ | 82,792 KB |
実行使用メモリ | 292,368 KB |
最終ジャッジ日時 | 2024-09-29 03:19:31 |
合計ジャッジ時間 | 4,298 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 1 TLE * 1 -- * 23 |
ソースコード
from collections import deque n = int(input()) rosen = [[] for _ in range(n+1)] for i in range(n-1): a, b = map(int, input().split()) rosen[a].append(b) rosen[b].append(a) ans = set() def bfs(start): queue = deque([start]) d = [-1] * (n+1) d[start] = 0 while queue: v = queue.popleft() for i in rosen[v]: if d[i] != -1: continue d[i] = d[v] + 1 if d[i] <= 3: if start < i: ans.add((start, i)) else: ans.add((i, start)) queue.append(i) for i in range(1, n+1): bfs(i) print(len(ans))