Luogu P3393 逃离僵尸岛

https://www.luogu.org/problemnew/show/P3393

题目描述

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

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

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

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

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

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

输入输出格式

输入格式：

 

第一行4个整数(N,M,K,S)

第二行2个整数(P,Q)

接下来K行，ci，表示僵尸侵占的城市

接下来M行，ai,bi，表示一条无向边

 

输出格式：

 

一个整数表示最低花费

 

输入输出样例

输入样例#1： 复制

13 21 1 1
1000 6000
7
1 2
3 7
2 4
5 8
8 9
2 5
3 4
4 7
9 10
10 11
5 9
7 12
3 6
4 5
1 3
11 12
6 7
8 11
6 13
7 8
12 13

输出样例#1： 复制

11000

 

k个僵尸据点，如果跑k次dijkstra必定超时

这个时候虚点站了出来，瞬间把k次降成了1次（就不能考虑一下将臣的感受QAQ）

构造一个0号节点，把它与k个僵尸据点连起来，跑一边dijkstra就可以知道所有点到k个僵尸据点的最短距离

再跑一边dijkstra

注意若变量开了longlong，堆，输出也要longlong

#include<cstdio>
#include<queue>
#define ll long long
using namespace std;

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

const int N=1e6+4;
int n,m,k,s,p,h;
ll d[N],a[N];
bool fl[N];
int cnt,to[N],nxt[N],he[N],w[N];
struct A{int id; ll x; };

bool operator >(A i,A j)
{
	return i.x>j.x;
}

priority_queue<A,vector<A>,greater<A> >q;

inline void add(int u,int v,int k)
{
	to[++cnt]=v,nxt[cnt]=he[u],he[u]=cnt,w[cnt]=k;
}

int main()
{
	n=read(),m=read(),k=read(),s=read();
	p=read(),h=read();
	for(int i=1;i<=k;i++) 
	{
		int u=read();	add(0,u,0);  a[u]=1e18;
	}
	for(int i=1;i<=m;i++) 
	{
		int u=read(),v=read();
		add(u,v,1),add(v,u,1);
	}
	for(int i=0;i<=n;i++) d[i]=2e9;
	q.push((A){0,d[0]=0});
	while(!q.empty()) 
	{
		while(!q.empty()&&fl[q.top().id]) q.pop();
		if(q.empty()) break;
		int u=q.top().id; q.pop(); fl[u]=1;
		for(int e=he[u];e;e=nxt[e])
		{
			int v=to[e];
			if(!fl[v]&&d[v]>d[u]+w[e]) 
				q.push((A){v,d[v]=d[u]+w[e]});
		}
	}
	for(int i=1;i<=n;i++) 
	{
		if(!a[i]) a[i]=(d[i]<=s)?h:p;
		d[i]=1e18,fl[i]=0;
	}
	q.push((A){1,d[1]=0});
	while(!q.empty())
	{
		while(!q.empty()&&fl[q.top().id]) q.pop();
		if(q.empty()) break;
		int u=q.top().id; q.pop(),fl[u]=1;
		for(int e=he[u];e;e=nxt[e])
		{
			int v=to[e];
			if(!fl[v]&&d[v]>d[u]+a[v]) 
				q.push((A){v,d[v]=d[u]+a[v]});
		}
	}
	printf("%lld\n",d[n]-a[n]);
	return 0;
}