結果

問題 No.806 木を道に
ユーザー betrue12
提出日時 2019-03-22 21:27:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,519 bytes
コンパイル時間 2,423 ms
コンパイル使用メモリ 202,780 KB
最終ジャッジ日時 2025-01-06 23:43:16
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 9 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

struct Diameter{
    vector<int> center;
    int N;

    pair<vector<int>, vector<int>> calc_dist(int start, const vector<int> G[]){
        vector<int> dist(N, -1), prev(N);
        dist[start] = 0;
        prev[start] = -1;
        queue<int> que;
        que.push(start);
        while(que.size()){
            int i = que.front(); que.pop();
            for(auto j : G[i]){
                if(dist[j] == -1){
                    dist[j] = dist[i]+1;
                    prev[j] = i;
                    que.push(j);
                }
            }
        }
        return {dist, prev};
    }

    int calculate(int n, const vector<int> G[]){
        N = n;
        auto res1 = calc_dist(0, G);
        auto d1 = res1.first;
        int v1 = max_element(d1.begin(), d1.end()) - d1.begin();
        auto res2 = calc_dist(v1, G);
        auto d2 = res2.first;
        auto p2 = res2.second;
        int v2 = max_element(d2.begin(), d2.end()) - d2.begin();
        int diam = d2[v2];
        while(v2 != -1){
            if(d2[v2] == diam/2 || d2[v2] == (diam+1)/2) center.push_back(v2);
            v2 = p2[v2];
        }
        return diam;
    }
};

vector<int> edges[100000];

int main(){
    int N;
    cin >> N;
    for(int i=0; i<N-1; i++){
        int a, b;
        cin >> a >> b;
        edges[a-1].push_back(b-1);
        edges[b-1].push_back(a-1);
    }

    Diameter diam;
    int d = diam.calculate(N, edges);
    cout << (N-1) - d << endl;
    return 0;
}
0