Luogu P4568 [JLOI2011]飞行路线

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

题目描述

Alice和Bob现在要乘飞机旅行，他们选择了一家相对便宜的航空公司。该航空公司一共在nn个城市设有业务，设这些城市分别标记为00到n-1n−1，一共有mm种航线，每种航线连接两个城市，并且航线有一定的价格。

Alice和Bob现在要从一个城市沿着航线到达另一个城市，途中可以进行转机。航空公司对他们这次旅行也推出优惠，他们可以免费在最多kk种航线上搭乘飞机。那么Alice和Bob这次出行最少花费多少？

输入输出格式

输入格式：

 

数据的第一行有三个整数，n,m,kn,m,k，分别表示城市数，航线数和免费乘坐次数。
第二行有两个整数，s,ts,t，分别表示他们出行的起点城市编号和终点城市编号。
接下来有m行，每行三个整数，a,b,ca,b,c，表示存在一种航线，能从城市aa到达城市bb，或从城市bb到达城市aa，价格为cc。

 

输出格式：

 

只有一行，包含一个整数，为最少花费。

 

输入输出样例

输入样例#1： 复制

5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100

输出样例#1： 复制

8

说明

对于30%的数据,2 \le n \le 50,1 \le m \le 300,k=02≤n≤50,1≤m≤300,k=0;
对于50%的数据,2 \le n \le 600,1 \le m \le 6000,0 \le k \le 12≤n≤600,1≤m≤6000,0≤k≤1;
对于100%的数据,2 \le n \le 10000,1 \le m \le 50000,0 \le k \le 102≤n≤10000,1≤m≤50000,0≤k≤10,0 \le s,t<n,0 \le a,b<n,a\neq b,0 \le c \le 10000≤s,t<n,0≤a,b<n,a≠b,0≤c≤1000

 

分层dijkstra

d[v][j]=min(d[u][j]+w[e]，d[u][j+1])

#include<cstdio>
#include<queue>
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+5;
int n,m,K,s,t,d[N][11],ans;
bool fl[N][11];
int cnt,to[N],nxt[N],he[N],w[N];

struct A{ int id1,id2,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(),t=read();
	for(int i=1;i<=m;i++)
	{
		int u=read(),v=read(),k=read();
		add(u,v,k),add(v,u,k);
	}
	for(int i=0;i<n;i++)
		for(int j=0;j<=K;j++) d[i][j]=2e9;
	q.push((A){s,K,d[s][K]=0}); 
	while(!q.empty())
	{
		while(!q.empty()&&fl[q.top().id1][q.top().id2]) q.pop();
		if(q.empty()) break;
		int u=q.top().id1,x=q.top().id2;
		fl[u][x]=1; q.pop();
		for(int e=he[u];e;e=nxt[e])
		{
			int v=to[e];
			if(!fl[v][x]&&d[v][x]>d[u][x]+w[e]) 
				q.push((A){v,x,d[v][x]=d[u][x]+w[e]});
			if(x>=1&&!fl[v][x]&&d[v][x-1]>d[u][x])
				q.push((A){v,x-1,d[v][x-1]=d[u][x]});
		}
	}
	ans=2e9;
	for(int i=0;i<=K;i++) ans=min(ans,d[t][i]);
	printf("%d\n",ans);
	return 0;
}