P4568 [JLOI2011] 飞行路线
分层图能够解决的问题是有边权为0的情况,比如说你可以将k条边变为零,然后求最小距离,然后我们就可以用分层图的思想
其实就是在转移过程中会涉及层内转移和层间转移,层内转移就是正常的最短路,而层间转移就是加一个条件限制,然后进行转移
在飞行路线题中,我们只需要使用dist[i][j]表示到达第i个点,并且使用了j次免费机会的最短路径,假设当前点是a,另一个点为b,则dist[a][j]可以转移到dist[b][j]与dist[b][j+1],所以跑一遍最短路即可
// Problem: P4568 [JLOI2011] 飞行路线
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P4568
// Memory Limit: 125 MB
// Time Limit: 1000 ms
#include<iostream>
#include<cstring>
#include<string>
#include<sstream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<map>
#include<stack>
#include<vector>
#include<set>
#include<unordered_map>
#include<ctime>
#include<cstdlib>
#define fi first
#define se second
using namespace std;
typedef long long ll;
typedef double db;
typedef pair<int,int> PII;
typedef pair<int,pair<int,int> > PIII;
const double eps=1e-7;
const int N=5e5+7 ,M=5e5+7, INF=0x3f3f3f3f,mod=1e9+7;
const long long int llINF=0x3f3f3f3f3f3f3f3f;
inline ll read() {ll x=0,f=1;char c=getchar();while(c<'0'||c>'9') {if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') {x=(ll)x*10+c-'0';c=getchar();} return x*f;}
inline void write(ll x) {if(x < 0) {putchar('-'); x = -x;}if(x >= 10) write(x / 10);putchar(x % 10 + '0');}
inline void write(ll x,char ch) {write(x);putchar(ch);}
void stin() {freopen("in_put.txt","r",stdin);freopen("my_out_put.txt","w",stdout);}
bool cmp0(int a,int b) {return a>b;}
template<typename T> T gcd(T a,T b) {return b==0?a:gcd(b,a%b);}
template<typename T> T lcm(T a,T b) {return a*b/gcd(a,b);}
void hack() {printf("\n----------------------------------\n");}
int T,hackT;
int n,m,k;
int h[N],e[M],ne[M],w[M],idx;
int s,t;
int dist[N][11];
bool st[N][11];
struct Node{
int d,ver,cnt;
bool operator > (const Node &k) const {
return d>k.d;
}
};
void add(int a,int b,int c) {
e[idx]=b,ne[idx]=h[a],w[idx]=c,h[a]=idx++;
}
void dijkstra(int u) {
memset(dist,0x3f,sizeof dist);
dist[u][0]=0;
priority_queue<Node,vector<Node>,greater<Node> > q;
q.push({dist[u][0],u,0});
while(q.size()) {
auto it=q.top();
q.pop();
int ver=it.ver;
if(st[ver][it.cnt]) continue;
st[ver][it.cnt]=true;
for(int i=h[ver];i!=-1;i=ne[i]) {
int j=e[i];
if(dist[j][it.cnt]>dist[ver][it.cnt]+w[i]) {
dist[j][it.cnt]=dist[ver][it.cnt]+w[i];
q.push({dist[j][it.cnt],j,it.cnt});
}
if(it.cnt+1<=k&&dist[j][it.cnt+1]>dist[ver][it.cnt]) {
dist[j][it.cnt+1]=dist[ver][it.cnt];
q.push({dist[j][it.cnt+1],j,it.cnt+1});
}
}
}
}
void solve() {
n=read(),m=read(),k=read();
memset(h,-1,sizeof h);
s=read()+1,t=read()+1;
while(m--) {
int a=read()+1,b=read()+1,c=read();
add(a,b,c),add(b,a,c);
}
dijkstra(s);
int ans=INF;
for(int i=0;i<=k;i++) ans=min(ans,dist[t][i]);
printf("%d\n",ans);
}
int main() {
// init();
// stin();
// scanf("%d",&T);
T=1;
while(T--) hackT++,solve();
return 0;
}