結果

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

ソースコード

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 += sum(x != v and x != u for x in adj_list[w])
    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