分层图+Dijkstra

#include <iostream>
#include <queue>
using namespace std;
typedef long long ll;

const int MAXN=100100;
const int MAXM=500100;
const int INF=0x3f3f3f3f;

struct Edge
{
    int to,next;
    int w;
}e[MAXM*80];
int head[MAXM*80],cnt=0;
void addEdge(int x,int y,int z)
{
    cnt++;
    e[cnt].to=y;
    e[cnt].next=head[x];
    e[cnt].w=z;
    head[x]=cnt;
    return;
}

int n,m,k;
ll t[MAXN*80];
bool vis[MAXN*80];

void Dijkstra()
{
    for (int i=1;i<=(k+1)*n;i++)
    {
        t[i]=INF;
        vis[i]=false;
    }
    t[1]=0;

    priority_queue< pair<int,int> >pq;
    pq.push(make_pair(0,1));

    while (!pq.empty())
    {
        int cur=pq.top().second;
        pq.pop();

        if (vis[cur])   continue;
        vis[cur]=true;

        for (int i=head[cur];i!=0;i=e[i].next)
        {
            if (t[e[i].to]>t[cur]+e[i].w)
            {
                t[e[i].to]=t[cur]+e[i].w;
                pq.push(make_pair(-t[e[i].to],e[i].to));
            }
        }
    }
    return;
}

int main()
{
    cin>>n>>m>>k;

    int u,v,w;
    for (int i=1;i<=m;i++)
    {
        cin>>u>>v>>w;
        for (int i=0;i<=k;i++)
        {
            addEdge(u+i*n,v+i*n,w);
            addEdge(v+i*n,u+i*n,w);
        }
        for (int i=0;i<k;i++)
        {
            addEdge(u+i*n,v+(i+1)*n,0);
            addEdge(v+i*n,u+(i+1)*n,0);
        }
    }

    Dijkstra();

    ll ans=INF;
    for (int i=1;i<=(k+1);i++)
    {
        ans=min(ans,t[i*n]);
    }
    cout<<ans<<endl;

    return 0;
}