BZOJ2763飞行路线——分层图+Dij堆优化

Description
Alice和Bob现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务，设这些城市分别标记为0到n-1，一共有m种航线，每种航线连接两个城市，并且航线有一定的价格。Alice和Bob现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多k种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少？
Input
数据的第一行有三个整数，n,m,k，分别表示城市数，航线数和免费乘坐次数。
第二行有两个整数，s,t，分别表示他们出行的起点城市编号和终点城市编号。(0<=s,t<n)
接下来有m行，每行三个整数，a,b,c，表示存在一种航线，能从城市a到达城市b，或从城市b到达城市a，价格为c。(0<=a,b<n,a与b不相等，0<=c<=1000)

Output

只有一行，包含一个整数，为最少花费。
Sample Input
5 6 1

0 4

0 1 5

1 2 5

2 3 5

3 4 5

2 3 3

0 2 100

Sample Output
8
HINT

对于30%的数据,2<=n<=50,1<=m<=300,k=0;

对于50%的数据,2<=n<=600,1<=m<=6000,0<=k<=1;

对于100%的数据,2<=n<=10000,1<=m<=50000,0<=k<=10.

---

可以免费$k$次，所以我们可以建$k+1$张图，每张图都是原图的复制。

那么对于点$x$，点$x+k\ast n$都是与其具有相同性质的点。（$n$是原图总点数），如果$x$到$y$有一条航线，那么我们在原图上连上一条从$x$到$y+n$的单向边（也就是当前图的点$x$，到下一层图里的$y$点），这条边的权值为$0$，也就代表这条航线是免费的。

那么我们从$S$开始跑，最后的答案就是$dis[T+k\ast n]$。向上爬了$k$层图，也就是走了$k$条免费的边。

---

然后就是这道题出题人卡SPFA，就算能卡常卡过去，也是磕磕绊绊的。

而Dijkstra+堆优化就能完美快速地跑过这道题目了。

#pragma GCC optimize(3)
#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define MAXN 2700005
using namespace std;
char tc(){ 
	static char fl[100000],*A=fl,*B=fl; 
	return A==B&&(B=(A=fl)+fread(fl,1,100000,stdin),A==B)?EOF:*A++; } 
int read(){ 
	char c;while(c=tc(),(c<'0'||c>'9')&&c!='-'); 
	int x=0,y=1;c=='-'?y=-1:x=c-'0'; 
	while(c=tc(),c>='0'&&c<='9')x=x*10+c-'0'; 
	return x*y; 
}
int n,m,k,S,T,cnt,head[MAXN],nxt[MAXN],vis[MAXN],dis[MAXN];
struct node{
	int to,val;
}L[MAXN];
void add(int x,int y,int c){
	L[cnt]=(node){y,c};
	nxt[cnt]=head[x];head[x]=cnt;cnt++;
}
struct edge{
	int to,num;
	bool operator>(const edge x)const{
		return num>x.num;
	}
};
priority_queue<edge,vector<edge>,greater<edge> > q;
int main()
{
	n=read();m=read();k=read();
	memset(dis,63,sizeof(dis));
	memset(head,-1,sizeof(head));
	S=read()+1;T=read()+1;dis[S]=0;
	for(int i=1;i<=m;i++){
		int x=read()+1,y=read()+1,c=read();
		for(int j=0;j<=k;j++)
		  add(x+j*n,y+j*n,c),add(y+j*n,x+j*n,c);
	}
	for(int x=1;x<=k*n;x++)
	 for(int i=head[x];i!=-1;i=nxt[i]){
	 	int to=L[i].to;
	 	add(x,to+n,0);
	 }
	q.push((edge){S,0});
	while(!q.empty()){
		edge x=q.top();q.pop();
		if(vis[x.to]) continue;vis[x.to]=1;
		for(int i=head[x.to];i!=-1;i=nxt[i]){
			int to=L[i].to;
			if(!vis[to]&&dis[x.to]+L[i].val<dis[to]){
				dis[to]=dis[x.to]+L[i].val;
				q.push((edge){to,dis[to]});
			}
		}
	}
	printf("%d",dis[T+k*n]);
	return 0;
}