結果

問題 No.2427 Tree Distance Two
ユーザー loop0919loop0919
提出日時 2023-08-18 22:02:37
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 629 bytes
コンパイル時間 212 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 509,836 KB
最終ジャッジ日時 2024-11-28 07:10:16
合計ジャッジ時間 39,920 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 46 ms
59,136 KB
testcase_01 AC 44 ms
384,980 KB
testcase_02 AC 45 ms
59,136 KB
testcase_03 AC 1,992 ms
477,812 KB
testcase_04 AC 42 ms
58,752 KB
testcase_05 TLE -
testcase_06 AC 42 ms
59,136 KB
testcase_07 TLE -
testcase_08 TLE -
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 AC 585 ms
169,636 KB
testcase_15 AC 47 ms
54,016 KB
testcase_16 AC 50 ms
60,544 KB
testcase_17 AC 46 ms
53,632 KB
testcase_18 AC 52 ms
60,416 KB
testcase_19 AC 44 ms
53,632 KB
testcase_20 AC 149 ms
79,360 KB
testcase_21 AC 148 ms
79,872 KB
testcase_22 AC 123 ms
77,696 KB
testcase_23 AC 116 ms
77,204 KB
testcase_24 AC 155 ms
79,744 KB
testcase_25 AC 511 ms
102,016 KB
testcase_26 AC 1,260 ms
136,960 KB
testcase_27 AC 861 ms
120,468 KB
testcase_28 AC 1,837 ms
166,272 KB
testcase_29 AC 1,735 ms
155,264 KB
testcase_30 AC 1,249 ms
135,168 KB
testcase_31 AC 795 ms
114,944 KB
testcase_32 AC 1,198 ms
134,912 KB
testcase_33 AC 1,025 ms
128,076 KB
testcase_34 AC 309 ms
471,876 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
N = int(input())

tree = [set() for _ in range(N)]

for i in range(N-1):
    a, b = map(lambda x: int(x) - 1, input().split())
    tree[a].add(b)
    tree[b].add(a)

def bfs(a):
    visited = set([a])
    que = deque([(a, 0)])
    cnt = 0
    
    while que:
        now, dist = que.popleft()
        if dist == 2:
            cnt += 1
        if dist > 2:
            break
        
        for next in tree[now]:
            if next in visited:
                continue
            visited.add(next)
            que.append((next, dist+1))
    
    return cnt


for i in range(N):
    print(bfs(i))
0