BZOJ_1598_[Usaco2008 Mar]牛跑步_A*

Posted 2020-11-07 fcwww

BZOJ_1598_[Usaco2008 Mar]牛跑步_A*

Description

BESSIE准备用从牛棚跑到池塘的方法来锻炼. 但是因为她懒,她只准备沿着下坡的路跑到池塘, 然后走回牛棚. BES

SIE也不想跑得太远,所以她想走最短的路经. 农场上一共有M (1 <= M <= 10,000)条路, 每条路连接两个用1..N(1

 <= N <= 1000)标号的地点. 更方便的是,如果X>Y,则地点X的高度大于地点Y的高度. 地点N是BESSIE的牛棚;地点1

是池塘. 很快, BESSIE厌倦了一直走同一条路.所以她想走不同的路,更明确地讲,她想找出K (1 <= K <= 100)条不

同的路经.为了避免过度劳累,她想使这K条路经为最短的K条路经. 请帮助BESSIE找出这K条最短路经的长度.你的程

序需要读入农场的地图, 一些从X_i到Y_i 的路经和它们的长度(X_i, Y_i, D_i). 所有(X_i, Y_i, D_i)满足(1 <=

 Y_i < X_i; Y_i < X_i <= N, 1 <= D_i <= 1,000,000).

Input

* 第1行: 3个数: N, M, 和K

* 第 2..M+1行: 第 i+1 行包含3个数 X_i, Y_i, 和 D_i, 表示一条下坡的路.

Output

* 第1..K行: 第i行包含第i最短路经的长度,或-1如果这样的路经不存在.

如果多条路经有同样的长度,请注意将这些长度逐一列出.

Sample Input

5 8 7
5 4 1
5 3 1
5 2 1
5 1 1
4 3 4
3 1 1
3 2 1
2 1 1

Sample Output

1
2
2
3
6
7
-1

---

 

需要输出前K短路的长度，这里使用A*算法。

 

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <ext/pb_ds/priority_queue.hpp>
#define RR register
inline char nc() {
    static char buf[100000],*p1,*p2;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd() {
    register int x=0;
    register char s=nc();
    while(s<‘0‘||s>‘9‘) s=nc();
    while(s>=‘0‘&&s<=‘9‘) x=(x<<3)+(x<<1)+s-‘0‘,s=nc();
    return x;
}
using namespace std;
using namespace __gnu_pbds;
#define N 10050
#define M 100050
map <pair<int,int> ,int>mp;
int head[N],to[M],nxt[M],cnt,val[M],n,m,K,dis[N],vis[N],now,xx[M],yy[M],zz[M],len;
__attribute__((optimize("-O2")))inline void add(int u,int v,int w) {
    to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w;
}
__attribute__((optimize("-O2")))struct node {
    int v,p;
    node() {}
    node(int v_,int p_):
        v(v_),p(p_) {}
    inline bool operator < (const node &x) const {
        /*if(p==n&&x.p!=n) return 1;
        if(x.p==n&&p!=n) return 0;*/
        return v+dis[p]>x.v+dis[x.p];
    }
};
__gnu_pbds::priority_queue<pair<int,int> >q1;
__gnu_pbds::priority_queue<node>q2;
__attribute__((optimize("-O2")))void dij() {
    memset(dis,0x3f,sizeof(dis));
    dis[n]=0; q1.push(make_pair(0,n));
    while(!q1.empty()) {
        int x=q1.top().second;q1.pop();
        int i;
        if(vis[x]) continue;
        vis[x]=1;
        for(i=head[x];i;i=nxt[i]) {
            if(dis[to[i]]>dis[x]+val[i]) {
                dis[to[i]]=dis[x]+val[i];
                q1.push(make_pair(-dis[to[i]],to[i]));
            }
        }
    }
}
__attribute__((optimize("-O2")))void A_xing() {
    q2.push(node(0,1));
    while(!q2.empty()) {
        node t=q2.top(); q2.pop();
        RR int x=t.p,i,v=t.v;
        if(x==n) {
            now++; len=v;
            printf("%d\n",v); 
            if(now==K) exit(0);
        }
        for(i=head[x];i;i=nxt[i]) {
            int w=v+val[i];
            /*if(!mp.count(make_pair(w,to[i]))) {
                mp[make_pair(w,to[i])]=1;
                q2.push(node(w,to[i]));
            }*/
            q2.push(node(w,to[i]));
        }
    }
}
__attribute__((optimize("-O2")))int main() {
    n=rd(); m=rd(); K=rd();
    RR int i,x,y,z;
    for(i=1;i<=m;i++) {
        xx[i]=rd(); yy[i]=rd(); zz[i]=rd();
        add(xx[i],yy[i],zz[i]);
    }
    dij();
    memset(head,0,sizeof(head)); cnt=0;
    for(i=1;i<=m;i++) {
        add(yy[i],xx[i],zz[i]);
    }
    A_xing();
    K-=now;
    while(K--) puts("-1");
}