最短路【洛谷P3393】 逃离僵尸岛

P3393 逃离僵尸岛

题目描述

小a住的国家被僵尸侵略了！小a打算逃离到该国唯一的国际空港逃出这个国家。

该国有N个城市，城市之间有道路相连。一共有M条双向道路。保证没有自环和重边。

K个城市已经被僵尸控制了，如果贸然闯入就会被感染TAT...所以不能进入。由其中任意城市经过不超过S条道路就可以到达的别的城市，就是危险城市。换句话说只要某个没有被占城市到某个被占城市不超过s距离，就是危险。

小a住在1号城市，国际空港在N号城市，这两座城市没有被侵略。小a走每一段道路（从一个城市直接到达另外一个城市）得花一整个白天，所以晚上要住旅店。安全的的城市旅馆比较便宜要P元，而被危险的城市，旅馆要进行安保措施，所以会变贵，为Q元。所有危险的城市的住宿价格一样，安全的城市也是。在1号城市和N城市，不需要住店。

小a比较抠门，所以他希望知道从1号城市到N号城市所需要的最小花费。

输入数据保证存在路径，可以成功逃离。输入数据保证他可以逃离成功。

最短路。

判断最后一个点是否安全。

code：

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

#define int long long

using namespace std;

const int wx=200017;

inline int read(){
    int sum=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
    return sum*f;
}

int head[wx],dis[wx],avis[wx],vis[wx],flag[wx];
int num,n,m,p,Q,s,k;

struct e{
    int nxt,to;
}edge[wx*2];

void add(int from,int to){
    edge[++num].nxt=head[from];
    edge[num].to=to;
    head[from]=num;
}

struct node{
    int u,d;
    friend bool operator < (const node & a,const node & b){
        return a.d>b.d;
    }
};

queue<int > q;
void bfs(){
    while(q.size()){
        int u=q.front(); q.pop();
        for(int i=head[u];i;i=edge[i].nxt){
            int v=edge[i].to;
            if(dis[u]+1<=s&&dis[v]>dis[u]+1&&!avis[v]&&!flag[v]){
                dis[v]=dis[u]+1; flag[v]=1;
                q.push(v);
            }
        }
    }
}

priority_queue<node > pq;

void Dij(){
    for(int i=1;i<=n;i++)dis[i]=214748364222222LL,vis[i]=0;
    pq.push((node){1,0}); dis[1]=0;
    while(pq.size()){
        int u=pq.top().u; pq.pop();
        if(vis[u])continue; vis[u]=1;
        for(int i=head[u];i;i=edge[i].nxt){
            int v=edge[i].to;
            if(avis[v])continue;
            if(flag[v]){
                if(dis[v]>dis[u]+Q){
                    dis[v]=dis[u]+Q;
                    pq.push((node){v,dis[v]});
                }
            }
            else{
                if(dis[v]>dis[u]+p){
                    dis[v]=dis[u]+p;
                    pq.push((node){v,dis[v]});
                }
            }
        }
    }
}

signed main(){
    n=read(); m=read(); k=read(); s=read();
    p=read(); Q=read();
    memset(dis,0x3f,sizeof dis);
    for(int i=1;i<=k;i++){
        int x; x=read();
        q.push(x);avis[x]=1;dis[x]=0;
    }
    for(int i=1;i<=m;i++){
        int x,y;
        x=read(); y=read();
        add(x,y); add(y,x);
    }
    bfs();
    Dij();
    printf("%lld\n",dis[n]-(flag[n]?Q:p));
    return 0;
}