結果
| 問題 |
No.806 木を道に
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 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 |
ソースコード
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)