[JLOI2011] 飞行路线

水一道裸分层图。

洛谷 P4568 传送门

发现自己现在好虚啊，写这种裸题都得思考一下。

数组还开小了，110000开成了100000，导致第一遍提交RE，90分......

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

int n,m,k,s,t;
int hd[10005],to[100005],nx[100005],len[100005],ec;
int dis[110005],v[110005];

void edge(int af,int at,int el)
{
    to[++ec]=at;
    len[ec]=el;
    nx[ec]=hd[af];
    hd[af]=ec;
}

struct data
{
    int p,d;
    friend bool operator < (data q,data w)
    {
        return q.d>w.d;
    }
};

priority_queue<data>qq;

void dijkstra()
{
    memset(dis,0x3f,sizeof(dis));
    dis[s]=0;
    qq.push((data){s,0});
    while(!qq.empty())
    {
        data nw=qq.top();
        qq.pop();
        if(v[nw.p])continue;
        int pl=nw.p/n*n;
        for(int i=hd[nw.p%n];i;i=nx[i])
        {
            int dest=to[i]+pl;
            if((!v[dest])&&dis[dest]>nw.d+len[i])
            {
                dis[dest]=nw.d+len[i];
                qq.push((data){dest,dis[dest]});
            }
            dest=to[i]+pl+n;
            if(nw.p/n<k&&(!v[dest])&&dis[dest]>nw.d)
            {
                dis[dest]=nw.d;
                qq.push((data){dest,dis[dest]});
            }
        }
    }
}

int main()
{
    scanf("%d%d%d%d%d",&n,&m,&k,&s,&t);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        edge(u,v,w);
        edge(v,u,w);
    }
    dijkstra();
    int ans=0x3f3f3f3f;
    for(int i=k;i>=0;i--)ans=min(ans,dis[t+i*n]);
    printf("%d",ans);
    return 0;
}