洛谷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：

输出样例#1：

说明

对于20%数据，N<=50

对于100%数据，2 ≦ N ≦ 100000, 1 ≦ M ≦ 200000, 0 ≦ K ≦ N - 2, 0 ≦ S ≦ 100000

1 ≦ P ＜ Q ≦ 100000

今天真的是毫无状态啊，各种敲错变量看错题意，这么一道水题WA了六七次……

先DFS搜出每一个危险城市，再跑spfa就行。需要long long

之前还写了一个BFS版本的，WA掉就换了DFS（后来发现错因不在BFS）

BFS代码一并附在下面，那份代码没有加被感染城市不能走的判定，也没有用long long，当然WA咯，不过大致算法还是明确的。

 /**/

 #include<iostream>

 #include<cstdio>

 #include<cmath>

 #include<cstring>

 #include<algorithm>

 using namespace std;

 const long long inf=1000000000ll*1000000000ll+10ll;

 const int mxn=;

 struct edge{

     int v,nxt;

 }e[mxn];

 int hd[mxn],cnt=;

 void add_edge(int u,int v){

     e[++cnt].v=v;e[cnt].nxt=hd[u];hd[u]=cnt;

     return;

 }

 int n,m,k,s;

 long long P,Q;

 int kp[mxn];//被感染城市

 bool da[mxn];//危险标记

 int head=,tl=;

 int q[mxn];

 long long dis[mxn];

 bool vis[mxn];

 bool ban[mxn];

 void dfs(int u)

 {

     if(dis[u]<=s)da[u]=;

     if(dis[u]==s)return;

     for(int i=hd[u];i;i=e[i].nxt){

         int v=e[i].v;if(ban[v])continue;

         if(dis[v]>dis[u]+){

             dis[v]=dis[u]+;

             dfs(v);

         }

     }

     return;

 }

 bool inq[mxn];

 void SPFA(int x){

     for(int i=;i<=n;i++)dis[i]=inf;

     head=;tl=;

     q[head]=x;

     dis[x]=;

     inq[x]=;

     while(head!=tl+){

         int u=q[head];head=(head+)%;

         for(int i=hd[u];i;i=e[i].nxt){

             int v=e[i].v;long long cst=(da[v])?Q:P;

             if(ban[v])continue;

             if(dis[v]>dis[u]+cst){

                 dis[v]=dis[u]+cst;

                 if(!inq[v]){

                     tl=(tl+)%;

                     q[tl]=v;

                 }

             }

         }

         inq[u]=;

     }

     return;

 }

 int main(){

     scanf("%d%d%d%d",&n,&m,&k,&s);

     int i,j;

     scanf("%lld%lld",&P,&Q);

     int u,v;

     for(i=;i<=k;i++){

         scanf("%d",&kp[i]);

         ban[kp[i]]=;

     }

     for(i=;i<=m;i++){

         scanf("%d%d",&u,&v);

         add_edge(u,v);

         add_edge(v,u);

     }

     for(int i=;i<=n;i++)dis[i]=inf;

     for(i=;i<=k;i++){

         dis[kp[i]]=;

         dfs(kp[i]);

     }

     SPFA();

     if(da[n])dis[n]-=Q;

     else dis[n]-=P;

     printf("%lld\n",dis[n]);

     return ;

 }

 /**/

 #include<iostream>

 #include<cstdio>

 #include<cmath>

 #include<cstring>

 #include<algorithm>

 using namespace std;

 const int mxn=;

 struct edge{

     int v,nxt;

 }e[mxn];

 int hd[mxn],cnt=;

 void add_edge(int u,int v){

     e[++cnt].v=v;e[cnt].nxt=hd[u];hd[u]=cnt;

     return;

 }

 int n,m,k,s;

 int P,Q;

 int kp[mxn];//被感染城市

 bool da[mxn];//危险标记

 int head=,tl=;

 int q[mxn];

 int dis[mxn];

 bool vis[mxn];

 void BFS(int x){

     memset(vis,,sizeof vis);

     int i,j;

     head=;    tl=;

     q[tl]=x;vis[x]=;da[x]=;

     dis[x]=;

     int u,v;

     while(head!=tl+){

         u=q[head];head=(head+)%;

         for(i=hd[u];i;i=e[i].nxt){

             v=e[i].v;

             if(!vis[v]){

                 dis[v]=dis[u]+;

                 if(dis[v]<s){

                     tl=(tl+)%;

                     q[tl]=v;

                 }

                 da[v]=;

                 vis[v]=;

             }

         }

     }

     return;

 }

 bool inq[mxn];

 void SPFA(int x){

     memset(dis,0x3f,sizeof dis);

     head=;tl=;

     q[head]=x;

     dis[x]=;

     inq[x]=;

     while(head!=tl+){

         int u=q[head];head=(head+)%;

         for(int i=hd[u];i;i=e[i].nxt){

             int v=e[i].v;int cst=(da[v])?Q:P;

             if(dis[v]>dis[u]+cst){

                 dis[v]=dis[u]+cst;

                 if(!inq[v]){

                     tl=(tl+)%;

                     q[tl]=v;

                 }

             }

         }

         inq[u]=;

     }

     return;

 }

 int main(){

     scanf("%d%d%d%d",&n,&m,&k,&s);

     int i,j;

     scanf("%d%d",&P,&Q);

     int u,v;

     for(i=;i<=k;i++){

         scanf("%d",&kp[i]);

     }

     for(i=;i<=m;i++){

         scanf("%d%d",&u,&v);

         add_edge(u,v);

         add_edge(v,u);

     }

     for(i=;i<=k;i++){

         BFS(kp[i]);

     }

     SPFA();

     if(da[n])dis[n]-=Q;

     else dis[n]-=P;

     printf("%d\n",dis[n]);

     return ;

 }

BFS