首页 > 代码库 > bzoj 3575: [Hnoi2014]道路堵塞
bzoj 3575: [Hnoi2014]道路堵塞
Description
A 国有N座城市,依次标为1到N。同时,在这N座城市间有M条单向道路,每条道路的长度是一个正整数。现在,A国交通部指定了一条从城市1到城市N的路径, 并且保证这条路径的长度是所有从城市1到城市N的路径中最短的。不幸的是,因为从城市1到城市N旅行的人越来越多,这条由交通部指定的路径经常发生堵塞。 现在A国想知道,这条路径中的任意一条道路无法通行时,由城市1到N的最短路径长度是多少。
Input
输入文件第一行是三个用空格分开的正整数N、M和L,分别表示城市数目、单向道路数目和交通部指定的最短路径包含多少条道路。
按下来M行,每行三个用空格分开的整数a、b和c,表示存在一条由城市a到城市b的长度为c的单向道路。这M行的行号也是对应道路的编号,即其中第1行对
应的道路编号为1,第2行对应的道路编号为2,…,第M行对应的道路编号为M。最后一行为L个用空格分开的整数sp(1)…,,sp(L),依次表示从城
市1到城市N的由交通部指定的最短路径上的道路的编号。
Output
输出文件包含L行,每行为一个整数,第i行(i=1,2…,,L)的整数表示删去编号为sp(i)的道路后从城市1到城市N的最短路径长度。如果去掉后没有从城市1到城市N的路径,则输出一1。
真的是连SPFA都不会...
根据玄学猜想:删掉最短路上的一条边后的最短路一定是由三段组成:
最短路上的一段 1--x,非最短路上的一段x--y,以及最短路上的一段y--n;
然后对于每次删掉一条连接(u,v)的边我们就把u放入队列中,并强制不走删掉的那一条边,拿u来松弛其它点到1的最短路
如果走到了最短路上的某个点x,并且该点在最短路上的标号比u大(在u后面),把这个点放入堆中即可
这个我们只需要维护好两个数组,一个1到每个最短路上的点的距离,一个是每个最短路上的点到n的距离
之后就可以愉快的跑SPFA了
// MADE BY QT666 #include<cstdio> #include<algorithm> #include<cmath> #include<iostream> #include<queue> #include<cstring> using namespace std; typedef long long ll; const int N=700050; int gi() { int x=0,flag=1; char ch=getchar(); while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘) flag=-1;ch=getchar();} while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar(); return x*flag; } int head[N],to[N],nxt[N],c[N],cnt; int pre[N],pre2[N],id[N],to2[N],n,m,L; int a[N],vis[N],vis2[N],q[N*10],q2[N*10],dis[N]; struct data{ int id,dis; bool operator < (const data &a) const {return a.dis<dis;} }b[N]; priority_queue<data>Q; void lnk(int x,int y,int z){ to[++cnt]=y,c[cnt]=z,nxt[cnt]=head[x],head[x]=cnt; } void spfa(int x,int y,int z){ dis[x]=pre[id[x]];int t=0,sum=1,tail=0;q[0]=x;q2[++tail]=x; vis[x]=1; for(int i=1;i<=L+1;i++) vis2[to2[i]]=0; while(t<sum){ int now=q[t++];vis[now]=0; for(int i=head[now];i;i=nxt[i]){ if(i!=y){ int u=to[i]; if(id[u]>z){ if(!vis2[u]){ b[u].dis=dis[now]+c[i]+pre2[id[u]]; b[u].id=id[u];vis2[u]=1; q2[++tail]=u; } else b[u].dis=min(b[u].dis,dis[now]+c[i]+pre2[id[u]]); } else{ if(dis[u]>dis[now]+c[i]){ dis[u]=dis[now]+c[i]; if(!vis[u]) vis[u]=1,q[sum++]=u; } } } } } while(tail) Q.push(b[q2[tail]]),tail--; } int main(){ n=gi(),m=gi(),L=gi(); for(int i=1;i<=m;i++){ int x=gi(),y=gi(),z=gi(); lnk(x,y,z); } to2[1]=id[1]=1; for(int i=1;i<=L;i++){ a[i]=gi(); to2[i+1]=to[a[i]]; id[to[a[i]]]=i+1; } for(int i=1;i<=L;i++) pre[i+1]=pre[i]+c[a[i]]; for(int i=L;i>=1;i--) pre2[i]=pre2[i+1]+c[a[i]]; memset(dis,127,sizeof(dis)); for(int i=1;i<=L;i++){ spfa(to2[i],a[i],i); while(!Q.empty()&&Q.top().id<=i) Q.pop(); if(Q.empty()) puts("-1"); else printf("%d\n",Q.top().dis); } }
bzoj 3575: [Hnoi2014]道路堵塞