結果

問題 No.806 木を道に
ユーザー flippergoflippergo
提出日時 2020-12-30 19:10:35
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 870 bytes
コンパイル時間 462 ms
コンパイル使用メモリ 81,648 KB
実行使用メモリ 94,328 KB
最終ジャッジ日時 2024-10-07 22:12:48
合計ジャッジ時間 6,107 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 9 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque
N = int(input())
G = {i:[] for i in range(1,N+1)}
for _ in range(N-1):
    a,b = map(int,input().split())
    G[a].append(b)
    G[b].append(a)
d1 = [-1 for _ in range(N+1)]
d1[1] = 0
que = deque([(1,0)])
while que:
    x,d = que.popleft()
    for y in G[x]:
        if d1[y]<0:
            d1[y]=d+1
            que.append((y,d+1))
dmax = 0
for i in range(1,N+1):
    if dmax<d1[i]:
        dmax = d1[i]
        ind = i
d2 = [-1 for _ in range(N+1)]
d2[ind]=0
que = deque([(ind,0)])
P = [0 for _ in range(N+1)]
while que:
    x,d = que.popleft()
    for y in G[x]:
        if d2[y]<0:
            d2[y]=d+1
            P[y] = x
            que.append((y,d+1))
dmax = max(d2)
ind1 = d2.index(dmax)
ind0 = ind
ind = ind1
A = []
while ind!=ind0:
    ind = P[ind]
    A.append(ind)
A.pop()
cnt = 0
for a in A:
    cnt += len(G[a])-2
print(cnt)
0