飞行路线

分层图最短路

范围要开的足够大

注意：
add(u+j * n,v+j * n,w);
add(v+j * n,u+j * n,w);
add(u+(j-1) * n,v+j * n,0);
add(v+(j-1) * n,u+j * n,0);

最后终点之间连上边权为0的边

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
const int N=1000005;
int n,m,k,tot,s,t;
int head[N],dis[N];
bool vis[N];
struct node{
	int to,next,w;
}edge[N<<2];
void add(int u,int v,int w){
	edge[tot].to=v;
	edge[tot].next=head[u];
	edge[tot].w=w;
	head[u]=tot++;
}
void dij(int x){
	priority_queue<pair<int,int> > q;
	q.push(make_pair(0,x));
	dis[x]=0;
	while(!q.empty()){
		int u=q.top().second; q.pop();
		if(vis[u]) continue; vis[u]=1;
		for(int i=head[u];i!=-1;i=edge[i].next){
			int v=edge[i].to;
			if(dis[v]>dis[u]+edge[i].w){
				dis[v]=dis[u]+edge[i].w;
				q.push(make_pair(-dis[v],v));
			}
		}
	}
}
int main(){
	memset(dis,0x3f3f3f,sizeof(dis));
	memset(head,-1,sizeof(head));
	scanf("%d%d%d",&n,&m,&k);
	scanf("%d%d",&s,&t);
	for(int i=1;i<=m;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w); add(v,u,w);//di ling ceng
		for(int j=1;j<=k;j++){
//			add(u+(j-1)*n,u+j*n,0);
//			add(v+(j-1)*n,v+j*n,0);
			add(u+j*n,v+j*n,w);
			add(v+j*n,u+j*n,w);
			add(u+(j-1)*n,v+j*n,0);
			add(v+(j-1)*n,u+j*n,0);
		}
	}
	for(int i=1;i<=k;i++) add(t+(i-1)*n,t+i*n,0);
	dij(s);
	printf("%d",dis[t+k*n]);
	return 0;
}