BZOJ1598: [Usaco2008 Mar]牛跑步

题目链接

k短路裸题。直接上模板。

【代码】

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 1005
#define M 10005
#define INF 0x7fffffff
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,K,cnt,ans;
int b[M],p[N],nextedge[M],w[M];
int B[M],P[N],Next[M],W[M];
int Dis[N],dis[N];
int Flag[N];

struct node{
    int f,g,x;
    node(){}
    node(int ff,int gg,int xx){
        f=ff;g=gg;x=xx;
    }
    bool operator <(const node &a)const{
        return a.f<f;
    }
};

void Add(int x,int y,int z)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
}

void Ins(int x,int y,int z)
{
    B[cnt]=y;
    Next[cnt]=P[x];
    P[x]=cnt;
    W[cnt]=z;
}

void Input_Init()
{
    n=read(),m=read(),K=read();
    for(int i=1;i<=m;i++)
    {
        static int x,y,z;
        x=read(),y=read(),z=read();
        Add(x,y,z);Ins(y,x,z);
    }
}

void Spfa()
{
    queue<int>q;
    q.push(1);
    for(int i=2;i<=n;i++) Dis[i]=INF;
    while(!q.empty())
    {
        int k=q.front();q.pop();
        Flag[k]=0;
        for(int i=P[k];i;i=Next[i])
        {
            int v=B[i];
            if(Dis[v]>Dis[k]+W[i])
            {
                Dis[v]=Dis[k]+w[i];
                if(!Flag[v])
                {
                    Flag[v]=1;
                    q.push(v);
                }
            }
        }
    }
}

void Dij()
{
    priority_queue<node>q;
    q.push(node(Dis[n],0,n));
    for(int i=1;i<n;i++) dis[i]=INF;
    while(!q.empty())
    {
        node k=q.top();q.pop();
        if(Flag[k.x]==K) continue;Flag[k.x]++;
        if(k.x==1)
        {
            printf("%d\n",k.g);
            ans++;
        }
        for(int i=p[k.x];i;i=nextedge[i])
        {
            int v=b[i];
            q.push(node(Dis[v]+k.g+w[i],k.g+w[i],v));
        }
    }
    for(int i=ans+1;i<=K;i++) printf("%d\n",-1);
}

int main()
{
    Input_Init();
    Spfa();
    Dij();
    return 0;
}