寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

アルゴ式(beta版): C++による「グラフアルゴリズム:6 章:木の問題」- Q5. 木の直径 (2)

アルゴ式(beta版)の「グラフアルゴリズム:6 章:木の問題」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。

C++による「グラフアルゴリズム:6 章:木の問題」- Q5. 木の直径 (2)


僕が作成、提出したコードは、以下のとおりです。

Q5. 木の直径 (2)

algo-method.com

/*
アルゴ式(beta版): C++による「グラフアルゴリズム:6 章:木の問題」- Q5. 木の直径 (2)
Q5. 木の直径 (2)
https://algo-method.com/tasks/976
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
int main(void){
    ll n;
    cin>>n;
    vector<vector<ll>> g(n);
    for(ll i=0;i<n-1;i++){
        ll a,b;
        cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    ll r=0;
    vector<ll> d(n,-1);
    queue<ll> q;
    d[r]=0;
    q.push(r);
    while(q.size()>0){
        ll f=q.front();
        q.pop();
        for(auto i : g[f]){
            if(d[i]!=-1){
                continue;
            }
            d[i]=d[f]+1;
            q.push(i);
        }
    }
    ll num=max_element(d.begin(),d.end())-d.begin();
    vector<ll> v(n,-1);
    v[num]=0;
    q.push(num);
    while(q.size()>0){
        ll f=q.front();
        q.pop();
        for(auto i : g[f]){
            if(v[i]!=-1){
                continue;
            }
            v[i]=v[f]+1;
            q.push(i);
        }
    }
    cout<<*max_element(v.begin(),v.end())<<endl;
    return 0;
}

設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com