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