寝癖頭の解法

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

AtCoder Problems in C++ #E - ゾンビ島 (Zombie Island)

第15回日本情報オリンピック 予選(過去問)から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#E - ゾンビ島 (Zombie Island)

atcoder.jp

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

/*
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;
}