最短路【洛谷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;
}