分层图 P4822 [BJWC2012]冻结

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn=2e6+10;
const int maxm=5e6+10;
struct nod{ int next,to,val; }edge[maxn*10];
int head[maxn*10],cnt;
void add(int from,int to,int val){
    edge[++cnt]={ head[from],to,val };
    head[from]=cnt;
}
struct node{
    int u,d;
    bool operator<(const node&p)const{
        return d>p.d;
    }
};
int dis[maxn];
void dij(int s){
    priority_queue<node>q;
    q.push({s,0});
    memset(dis,0x3f,sizeof dis);
    dis[s]=0;
    while(!q.empty()){
        node temp=q.top();q.pop();
        if(dis[temp.u]!=temp.d)continue;
        for(int i=head[temp.u];i;i=edge[i].next){
            if(dis[edge[i].to]>dis[temp.u]+edge[i].val){
                dis[edge[i].to]=dis[temp.u]+edge[i].val;
                q.push({ edge[i].to,dis[edge[i].to] });
            }
        }
    }
}
int main()
{
    ios_base::sync_with_stdio(false);
    int n,m,k,s,t;
    cin>>n>>m>>k;s=1,t=n;
    for(int i=0,u,v,val;i<m;i++){
        cin>>u>>v>>val;
        add(u,v,val);
        add(v,u,val);
        for(int j=1;j<=k;j++){
            add(u+j*n-n,v+j*n,val/2);
            add(v+j*n-n,u+j*n,val/2);
            add(u+j*n,v+j*n,val);
            add(v+j*n,u+j*n,val);
        }
    }
    for(int i=1;i<=k;i++)
        add(t+i*n-n,t+i*n,0);
    dij(s);
    cout<<dis[t+k*n];
    return 0;
}