結果

問題 No.806 木を道に
コンテスト
ユーザー はむ吉🐹
提出日時 2019-03-22 22:06:09
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,241 bytes
コンパイル時間 87 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 50,528 KB
最終ジャッジ日時 2024-09-19 05:26:57
合計ジャッジ時間 7,379 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 9 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

import collections


INF = 10 ** 8


def breadth_first_search(adj_list, start_v):
    n = len(adj_list)
    visited = [False] * n
    pred = [None] * n
    dist = [INF] * n
    q = collections.deque()
    q.append(start_v)
    visited[start_v] = True
    dist[start_v] = 0
    while q:
        u = q.popleft()
        for v in adj_list[u]:
            if not visited[v]:
                visited[v] = True
                pred[v] = u
                dist[v] = dist[u] + 1
                q.append(v)
    return pred, dist


def solve(adj_list):
    n = len(adj_list)
    s = 0
    _, dist_a = breadth_first_search(adj_list, s)
    u = max(range(n), key=lambda x: dist_a[x])
    pred, dist_b = breadth_first_search(adj_list, u)
    v = max(range(n), key=lambda x: dist_b[x])
    path = [v]
    while path[-1] != u:
        path.append(pred[path[-1]])
    res = 0
    for w in path:
        if w != v and w != u:
            res += len(adj_list[w]) - 2
    return res


def main():
    n = int(input())
    adj_list = [set() for _ in range(n)]
    for _ in range(n - 1):
        a, b = (int(z) - 1 for z in input().split())
        adj_list[a].add(b)
        adj_list[b].add(a)
    print(solve(adj_list))


if __name__ == "__main__":
    main()
0