結果

問題 No.2638 Initial fare
ユーザー Cecil
提出日時 2024-02-19 22:16:08
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 666 bytes
コンパイル時間 213 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 51,052 KB
最終ジャッジ日時 2024-09-29 02:10:46
合計ジャッジ時間 4,089 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 1 TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(1 << 20)
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)
edges = set()

from collections import deque
def bfs(s):
    dist = [-1]*(N+1)
    dist[s] = 0
    que = deque()
    que.append(s)
    while que:
        v = que.popleft()
        for v2 in G[v]:
            if dist[v2] != -1 or dist[v2]>3:
                continue
            dist[v2] = dist[v]+1
            if dist[v2]-dist[s]<4:
                edges.add((min(v2,s),max(v2,s)))
            que.append(v2)
for i in range(1, N+1):
    bfs(i)
    #print(edges)
print(len(edges))
0