P4568 [JLOI2011] 飞行路线（分层图最短路模板题）

[JLOI2011] 飞行路线

题目描述

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

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

输入格式

第一行三个整数 \(n,m,k\)，分别表示城市数，航线数和免费乘坐次数。

接下来一行两个整数 \(s,t\)，分别表示他们出行的起点城市编号和终点城市编号。

接下来 \(m\) 行，每行三个整数 \(a,b,c\)，表示存在一种航线，能从城市 \(a\) 到达城市 \(b\)，或从城市 \(b\) 到达城市 \(a\)，价格为 \(c\)。

输出格式

输出一行一个整数，为最少花费。

样例 #1

样例输入 #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=0\)。

对于 \(50\%\) 的数据，\(2 \le n \le 600\)，\(1 \le m \le 6\times10^3\)，\(0 \le k \le 1\)。

对于 \(100\%\) 的数据，\(2 \le n \le 10^4\)，\(1 \le m \le 5\times 10^4\)，\(0 \le k \le 10\)，\(0\le s,t,a,b\le n\)，\(a\ne b\)，\(0\le c\le 10^3\)。

另外存在一组 hack 数据。

// Problem: P4568 [JLOI2011] 飞行路线
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P4568
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// Created Time: 2022-07-13 21:04:07
// 
// Powered by CP Editor (https://cpeditor.org)

//fw
#include<bits/stdc++.h>
#define zp ios::sync_with_stdio(false);cin.tie(0); cout.tie(0);
#define pii pair <int, int>
#define pll pair <ll, ll>
#define endl '\n'
#define il inline
#define pb push_back
#define fi first
#define se second
#define lc u<<1
#define rc u<<1|1
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=1e6+10,M=5e6+10;
int h[N],e[M],ne[M],w[M],idxn,m,k,idx,S,T,n;
bool st[N];
int dist[N];

void add(int a,int b,int c)
{
	e[idx]=b;
	w[idx]=c;
	ne[idx]=h[a];
	h[a]=idx++;
}
void dijkstra()
{
	memset(dist,0x3f,sizeof dist);
	dist[S]=0;

	priority_queue<pii,vector<pii>,greater<pii>>heap;
	heap.push({0,S});

	while(heap.size())
	{
		auto t=heap.top();
		heap.pop();

		int ver=t.second,dis=t.first;
		if(st[ver])continue;
		st[ver]=true;
		for(int i=h[ver];~i;i=ne[i])
		{
			int j=e[i];
			if(dist[j]>dis+w[i])
			{
				dist[j]=dis+w[i];
				heap.push({dist[j],j});
			}
		}
	}
}
int main()
{
	zp
	memset(h,-1,sizeof h);	
    cin>>n>>m>>k>>S>>T;
    while(m--)
    {
    	int a,b,c;
    	cin>>a>>b>>c;
    	add(a,b,c);//第一层建图
    	add(b,a,c);
    	for(int j=1;j<=k;j++)
    	{
    		add(a+(j-1)*n,b+j*n,0);//建立每层图之间的免费边，走该条边进入下一层相当于使用了一次免费机会
    		add(b+(j-1)*n,a+j*n,0);
    		
    		add(a+j*n,b+j*n,c);//每层图之间建边
    		add(b+j*n,a+j*n,c);
    		
    	}
    }


    dijkstra();
    int ans=0x3f3f3f3f;
    
  	for(int i=0;i<=k;i++)//由于到达终点时k次免费机会可能没用用尽，所以需要扫描一遍，也可以在每层图的终点间建立一条免费边
  	{
  		ans=min(ans,dist[T+i*n]);
  	}
  	cout<<ans<<endl;
    
    
    return 0;
}

2022/8/26 ps: 最近在更网络流24题的博客，做到了P4009 汽车加油行驶问题也是一道可以分层图最短路做的题