アルゴ式(beta版)の「グラフアルゴリズム:【補充】練習問題集」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。
コマを動かす
/* アルゴ式(beta版): C++による「グラフアルゴリズム:【補充】練習問題集」- コマを動かす コマを動かす https://algo-method.com/tasks/1007 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; int main(void){ ll n,s,k; cin>>n>>s>>k; vector<vector<ll>> g(n); for(ll i=0;i<n;i++){ ll a,b; cin>>a>>b; g[a].push_back(b); g[b].push_back(a); } vector<ll> d(n,-1); queue<ll> q; d[s]=0; q.push(s); while(q.size()>0){ ll v=q.front(); q.pop(); for(auto i : g[v]){ if(d[i]!=-1){ continue; } d[i]=d[v]+1; q.push(i); } } ll ans=0; for(ll i=0;i<n;i++){ if((d[i]<=k) && (d[i]%2==k%2)){ ans++; } } cout<<ans<<endl; return 0; }
設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com