luogu_P4822 [BJWC2012]冻结

#include<iostream> 
#include<cstdio>
#include<cstring>

#define ri register int
#define u int
#define NN 55*105
#define MM 1005*105

namespace fast {
    inline u in() {
        u x(0);
        char s=getchar();
        while(s<‘0‘||s>‘9‘) {
            s=getchar();
        }
        while(s>=‘0‘&&s<=‘9‘) {
            x=(x<<1)+(x<<3)+s-‘0‘;
            s=getchar();
        }
        return x;
    }
}

using fast::in;

namespace all {
    u cnt,N,M,K,h[NN];
    struct node {
        u to,next,va;
    } a[MM<<1]; 
    inline void add(const u &x,const u &y,const u &z) {
        a[++cnt].next=h[x],a[cnt].to=y,a[cnt].va=z,h[x]=cnt;
    }
    
    u d[NN],vt[NN],q[1000005];
    struct node1{
        u x,y,z;
    }b[MM];
    
    u run(const u &x,const u &n) {
        ri l(0),r(0),k;
        for(ri i(1);i<=n;++i) d[i]=0x7fffffff;
        d[x]=0,q[++r]=x;
        while(l<r) {
            k=q[++l],vt[k]=0;
            for(ri i=h[k]; i; i=a[i].next) {
                u y=a[i].to,z=a[i].va;
                if(d[k]+z<d[y]) {
                    d[y]=d[k]+z;
                    if(!vt[y]) {
                        vt[y]=1,q[++r]=y;
                    }
                }
            }
        }
    }
    
    inline void solve(){
        N=in(),M=in(),K=in();
        for(ri i(1);i<=M;++i){
            u _a(in()),_b(in()),_c(in());
            b[i].x=_a,b[i].y=_b,b[i].z=_c;
        }
        for(ri i(1);i<=K;++i){//建1-k层，普遍性地用循环建 
            for(ri j(1);j<=M;++j){
                u _a(b[j].x+(i-1)*N),_b(b[j].y+(i-1)*N),_c(b[j].z);
                add(_a,_b,_c),add(_b,_a,_c);//本层 
                add(_a,_b+N,_c/2),add(_b,_a+N,_c/2);//通往下一层 
            }
        }
        for(ri i(K+1);i<=K+1;++i){//建第k+1层，因为不想第1-k层还有下一层，所以单独来建 
            for(ri j(1);j<=M;++j){
                u _a(b[j].x+(i-1)*N),_b(b[j].y+(i-1)*N),_c(b[j].z);
                add(_a,_b,_c),add(_b,_a,_c);//只有本层 
            }
        }
        run(1,N*(K+1));//共k+1层， N*(K+1)个点 
        u ans(0x7fffffff);
        for(ri i(1);i<=K+1;++i){
            ans=std::min(d[N+(i-1)*N],ans);//所有层的第N点都是可以的 
        }
        printf("%d",ans);
    }
}

int main() {
    //freopen("x.txt","r",stdin);
    all::solve();
}