题目描述
Alice 和 Bob 现在要乘飞机旅行,他们选择了一家相对便宜的航空公司。该航空公司一共在n个城市设有业务,设这些城市分别标记为 0 到 n-1,一共有 m 种航线,每种航线连接两个城市,并且航线有一定的价格。
Alice 和 Bob 现在要从一个城市沿着航线到达另一个城市,途中可以进行转机。航空公司对他们这次旅行也推出优惠,他们可以免费在最多 kk 种航线上搭乘飞机。那么 Alice 和 Bob 这次出行最少花费多少?
输入格式
第一行三个整数 n,m,kn,m,k,分别表示城市数,航线数和免费乘坐次数。
接下来一行两个整数 s,ts,t,分别表示他们出行的起点城市编号和终点城市编号。
接下来 mm 行,每行三个整数a,b,ca,b,c,表示存在一种航线,能从城市 a 到达城市 b,或从城市 b 到达城市 a,价格为 c。
输出格式
输出一行一个整数,为最少花费。
输入样例
5 6 1
0 4
0 1 5
1 2 5
2 3 5
3 4 5
2 3 3
0 2 100
输出样例
8
分析:
分层图最短路:有 k次零代价通过一条路径,求总的最小花费。
我们可以把一个点拆成k+1个点,每个新结点代表使用不同次数免费通行后到达的原图结点。
这完全可以用一个结构体维护,然后跑最短路即可。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
const int N=2e5+5;
int n,m,k,S,T,idx;
int h[N],to[N],w[N],ne[N],vis[N][12],d[N][12];//d[i][j]是到i点使用j次免费的最小花费
struct node{
int no,dis,cnt;//cnt记录到达这个点使用的免费次数
node(){}
node(int no,int dis,int cnt):no(no),dis(dis),cnt(cnt){}
//dis小的先出
bool operator < (const node & t)const{
return dis>t.dis;
}
};
//链式前向星
void add(int x,int y,int z){
to[idx]=y,w[idx]=z,ne[idx]=h[x],h[x]=idx++;
to[idx]=x,w[idx]=z,ne[idx]=h[y],h[y]=idx++;
}
void dijkstra(int s){
priority_queue<node>pq;
memset(d,0x3f,sizeof d);
d[s][0]=0;
pq.push(node(s,0,0));
while(pq.size()){
node t=pq.top();
pq.pop();
int no=t.no,dis=t.dis,num=t.cnt;
//访问过的不再入队
if(vis[no][num]) continue;
vis[no][num]=1;
for(int i=h[no];~i;i=ne[i]){
int v=to[i],ww=w[i];
//可以免费通过
if(num<k&&d[v][num+1]>d[no][num]){
d[v][num+1]=d[no][num];//更新
pq.push(node(v,d[v][num+1],num+1));
}
//不可以免费通行
//注意:这里为什么不加else,因为no这个点可以不使用免费通行
if(d[v][num]>d[no][num]+ww){
d[v][num]=d[no][num]+ww;//更新
pq.push(node(v,d[v][num],num));
}
}
}
}
int main(){
cin>>n>>m>>k>>S>>T;
memset(h,-1,sizeof h);
while(m--){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
}
dijkstra(S);
int ans=0x3f3f3f3f;
//到终点使用0-k次免费的最小值
for(int i=0;i<=k;i++)
ans=min(ans,d[T][i]);
cout<<ans;
return 0;
}