题目描述 https://www.luogu.org/problem/P2939
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
priority_queue< pair<int,int> > q;
const int N=2e6+5,M=2e7+5;
int h[N],to[2*M],next[2*M],dis[2*M],tot;
int n,m,k,d[N],ans;
bool fl[N];
void add(int x,int y,int z)
{
to[++tot]=y;
dis[tot]=z;
next[tot]=h[x];
h[x]=tot;
}
void dij()
{
memset(d,0x3f,sizeof(d));
d[1]=0;
q.push(make_pair(0,1));
while(q.size())
{
int x=q.top().second;
q.pop();
if(fl[x]==1) continue;
fl[x]=1;
for(int i=h[x];i;i=next[i])
{
int y=to[i];
if(d[y]>d[x]+dis[i])
{
d[y]=d[x]+dis[i];
q.push(make_pair(-d[y],y));
}
}
}
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=m;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z),add(y,x,z);
for(int j=1;j<=k;j++)
{
add(n*j+x,n*j+y,z),add(n*j+y,n*j+x,z);
add(n*(j-1)+x,n*j+y,0),add(n*(j-1)+y,n*j+x,0);
}
}
dij();
ans=d[n];
for(int i=1;i<=k;i++) ans=min(ans,d[n*(i+1)]);
printf("%d",ans);
return 0;
}