分层图【p2939】[USACO09FEB]改造路Revamping Trails

Description

约翰一共有N)个牧场.由M条布满尘埃的小径连接.小径可 以双向通行.每天早上约翰从牧场1出发到牧场N去给奶牛检查身体.

通过每条小径都需要消耗一定的时间.约翰打算升级其中K条小径，使之成为高 速公路.在高速公路上的通行几乎是瞬间完成的，所以高速公路的通行时间为0.

请帮助约翰决定对哪些小径进行升级，使他每天从1号牧场到第N号牧场所花的时间最短

分层图最短路的裸题..

就不多BB了.

代码

#include<cstdio>
#include<queue>
#include<cstring>
#define R register
#define N 20008
using namespace std;
inline void in(int &x)
{
    int f=1;x=0;char s=getchar();
    while(s>'9' or s<'0'){if(s=='-')f=-1;s=getchar();}
    while(s>='0' and s<='9'){x=x*10+s-'0';s=getchar();}
    x*=f;
}
int head[N],tot,n,m,s,t,k;
int dis[N][25],ans=2147483647;
bool vis[N][25];
struct cod{int u,v,w;}edge[N*6+8];
inline void add(int x,int y,int z)
{
    edge[++tot].u=head[x];
    edge[tot].v=y;
    edge[tot].w=z;
    head[x]=tot;
}
struct coc{
    int u,d,used;
    bool operator <(const coc&a) const 
    {
        return d>a.d;
    }
};
inline void dijkstra()
{
    memset(dis,127,sizeof dis);
    dis[s][0]=0;
    priority_queue<coc>q;
    q.push((coc){s,0,0});
    while(!q.empty())
    {
        int u=q.top().u,now=q.top().used;
        q.pop();
        if(vis[u][now])continue;
        vis[u][now]=true;
        for(R int i=head[u];i;i=edge[i].u)
        {
            if(now<k and !vis[edge[i].v][now+1] and dis[edge[i].v][now+1]>dis[u][now])
            {
                dis[edge[i].v][now+1]=dis[u][now];
                q.push((coc){edge[i].v,dis[edge[i].v][now+1],now+1});
            }
            if(!vis[edge[i].v][now] and dis[edge[i].v][now]>dis[u][now]+edge[i].w)
            {
                dis[edge[i].v][now]=dis[u][now]+edge[i].w;
                q.push((coc){edge[i].v,dis[edge[i].v][now],now});
            }
        }
    }
}
int main()
{
    in(n),in(m),in(k);s=1;t=n;
    for(R int i=1,x,y,z;i<=m;i++)
    {
        in(x),in(y),in(z);
        add(x,y,z);add(y,x,z);
    }
    dijkstra();
    for(R int i=0;i<=k;i++)
        ans=min(ans,dis[t][i]);
    printf("%d",ans);
}