分层图最短路,注意数组范围
#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int N=2000005;
int n,m,tot,k;
bool vis[N];
int dis[N];
int head[N];
struct node{
int to,next,w;
}edge[N*10];
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;
dis[x]=0; q.push(make_pair(0,x));
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);
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);
for(int j=1;j<=k;j++){
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(n+(i-1)*n,n+n*i,0);
dij(1);
// for(int i=1;i<=n;i++) printf("--%d
",dis[i]);
printf("%d",dis[n+n*k]);
return 0;
}