[BZOJ] 2662: [BeiJing wc2012]冻结

https://www.lydsy.com/JudgeOnline/problem.php?id=2662

第一次写分层图(捂脸)

一开始真的naive地建图了，T到飞起..

可以省下建图的空间，直接在dis数组上拓展一维，同时维护点的编号和高度。

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

using namespace std;

inline int rd(){
    int ret=,f=;char c;
    while(c=getchar(),!isdigit(c))f=c=='-'?-:;
    while(isdigit(c))ret=ret*+c-'',c=getchar();
    return ret*f;
}

const int MAXN=*;

struct Edge{
    int next,to,w;
}e[];
int ecnt,head[MAXN];
inline void add(int x,int y,int w){
    e[++ecnt].next = head[x];
    e[ecnt].to = y;
    e[ecnt].w = w;
    head[x] = ecnt;
}

int n,m,num;

queue<int> Q,Qk;
int dis[MAXN][],inq[MAXN][];
void spfa(){
    memset(dis,0x3f,sizeof(dis));
    Q.push();Qk.push();dis[][]=;inq[][]=;
    while(!Q.empty()){
        int top=Q.front();Q.pop();int tmp=Qk.front();Qk.pop();
        inq[top][tmp]=;
        for(int i=head[top];i;i=e[i].next){
            int v=e[i].to;
            if(dis[v][tmp]>dis[top][tmp]+e[i].w){
                dis[v][tmp]=dis[top][tmp]+e[i].w;
                if(!inq[v][tmp]) Q.push(v),Qk.push(tmp),inq[v][tmp]=;
            }
            if(tmp<num&&dis[v][tmp+]>dis[top][tmp]+(e[i].w>>)){
                dis[v][tmp+]=dis[top][tmp]+(e[i].w>>);
                if(!inq[v][tmp+]) Q.push(v),Qk.push(tmp+),inq[v][tmp+]=;
            }
        }
    }
}
int main(){
    n=rd();m=rd();num=rd();
    int x,y,w;
    for(int i=;i<=m;i++){
        x=rd();y=rd();w=rd();
        add(x,y,w);add(y,x,w);
    }
    spfa();
    int ans=<<;
    for(int i=;i<=num;i++)ans=min(ans,dis[n][i]);
    cout<<ans;
    return ;
}