アルゴ式(beta版)の「グラフアルゴリズム:【補充】練習問題集」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。
二部グラフ判定 (木 ver.)
/* アルゴ式(beta版): C++による「グラフアルゴリズム:【補充】練習問題集」- 二部グラフ判定 (木 ver.) 二部グラフ判定 (木 ver.) https://algo-method.com/tasks/1006 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; void DFS(ll v,vector<vector<ll>> &g,vector<ll> &c,string &s){ for(auto i : g[v]){ if(c[i]!=-1){ if(c[i]==c[v]){ s="No\n"; } continue; } c[i]=1-c[v]; DFS(i,g,c,s); } } 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); } vector<ll> c(n,-1); string s="Yes\n"; for(ll i=0;i<n;i++){ if(c[i]!=-1){ continue; } c[i]=0; DFS(i,g,c,s); } cout<<s; return 0; }
木は常に二部グラフなので、"Yes"と出力するだけのソースコードでもACにはなると思うのですが、とにかく今は自分の腕を磨いているので、正攻法で解きました。
設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com