第15回日本情報オリンピック 予選(過去問)から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#E - ゾンビ島 (Zombie Island)
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #E - ゾンビ島 (Zombie Island) https://atcoder.jp/contests/joi2016yo/tasks/joi2016yo_e 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; ll D[1<<17],S[1<<17]; vector<ll> g[1<<17]; int main(void){ ll n,m,k,s,p,q; cin>>n>>m>>k>>s>>p>>q; queue<pair<ll,ll>> Q; for(ll i=0;i<n;i++){ D[i]=n; } for(ll i=0;i<k;i++){ ll c; cin>>c; c--; D[c]=0; Q.push({c,0}); } for(ll i=0;i<m;i++){ ll a,b; cin>>a>>b; a--; b--; g[a].push_back(b); g[b].push_back(a); } while(!(Q.empty())){ auto[u,d]=Q.front(); Q.pop(); if(d>D[u]){ continue; } for(ll v : g[u]){ if(D[v]>d+1){ D[v]=d+1; Q.push({v,d+1}); } } } priority_queue<pair<ll,ll>,vector<pair<ll,ll>>,greater<pair<ll,ll>>> pq; pq.push({0,0}); while(!(pq.empty())){ auto[d,u]=pq.top(); pq.pop(); if(S[u]){ continue; } S[u]=1; for(ll v : g[u]){ if(v==n-1){ cout<<d<<endl; return 0; } if(D[v]){ pq.push({d+((D[v]>s) ? p : q),v}); } } } return 0; }