[JLOI2011]飞行路线 (分层图,最短路)

Posted 2021-01-01 kv-stalin

题目链接

Solution

建立 (k+1) 层图跑 (Dijkstra) 就好了.

Code

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=200008;
int n,m,k,s,t;
struct sj{
    int to;
    int next;
    int w;
}a[maxn*10];
int head[maxn],size;
ll dis[maxn],dist[maxn];

void add(int x,int y,int w)
{
    a[++size].to=y;
    a[size].next=head[x];
    head[x]=size;
    a[size].w=w;
}

int read()
{
    char ch=getchar(); int f=1,w=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch<='9'&&ch>='0'){w=w*10+ch-'0';ch=getchar();}
    return f*w;
}

struct node {
    int u;ll d;
    bool operator <(const node& rhs) const {
        return d>rhs.d;
    }
};

inline void Dijkstra()
{
    memset(dis,127,sizeof(dis));
    dis[s]=0;
    priority_queue<node> q;
    q.push((node){s,0});
    while(!q.empty())
    {
        node xx=q.top(); q.pop();
        int u=xx.u,d=xx.d;
        if(d!=dis[u])continue;
        for(int i=head[u];i;i=a[i].next)
        {
            int tt=a[i].to,w=a[i].w;
            if(dis[u]+w<dis[tt])
            {
                dis[tt]=dis[u]+w;
                q.push((node){tt,dis[tt]});
            }
        }
    }
}

void pre(int x,int y,int w)
{
    for(int i=0;i<=k;i++)
    add(x+n*i,y+n*i,w),
    add(y+n*i,x+n*i,w);
    for(int i=0;i<k;i++)
    add(x+n*i,y+n*i+n,0),
    add(y+n*i,x+n*i+n,0);
}

int main()
{
    n=read(); m=read(); k=read();
    s=read(); t=read(); 
    for(int i=1;i<=m;i++)
    {
        int x,y,w;
        x=read();
        y=read();
        w=read();
        pre(x,y,w);
    }
    for(int i=0;i<=k;i++)
    add(t+i*n,n*(k+1)+1,0);
    Dijkstra();
    cout<<dis[n*(k+1)+1]<<endl;
}