洛谷P2901 [USACO08MAR]牛慢跑Cow Jogging

题目描述

Bessie has taken heed of the evils of sloth and has decided to get fit by jogging from the barn to the pond several times a week. She doesn't want to work too hard, of course, so she only plans to jog downhill to the pond and then amble back to the barn at her leisure.

Bessie also doesn't want to jog any too far either, so she generally takes the shortest sequence of cow paths to get to the pond. Each of the M (1 <= M <= 10,000) cow paths connects two pastures

conveniently numbered 1..N (1 <= N <= 1000). Even more conveniently, the pastures are numbered such that if X>Y then the cow path from pasture X to pasture Y runs downhill. Pasture N is the barn (at the top of the hill) and pasture 1 is the pond (at the bottom).

Just a week into her regimen, Bessie has begun to tire of always taking the same route to get to the pond. She would like to vary her route by taking different cow paths on different days. Specifically, Bessie would like to take exactly K (1 <= K <= 100) different routes for variety. To avoid too much exertion, she wants these to be the K shortest routes from the barn to the pond. Two routes are considered different if they comprise different sequences of cow paths.

Help Bessie determine how strenuous her workout will be by determining the lengths of each of the K shortest routes on the network of pastures. You will be supplied a list of downhill cow paths from X_i to Y_i along with the cow path's length: (X_i, Y_i, D_i) where (1 <= Y_i < X_i; Y_i < X_i <= N). Cowpath i has length D_i (1 <= D_i <= 1,000,000).

贝西尝到了懒惰的恶果——为了减肥，她不得不决定每周花几次时间在牛棚和池塘之间慢跑。但贝西并不想太累，所以她打算只跑从牛棚到池塘的下坡路，然后再慢慢地从池塘走回牛棚。 

同时，贝西也不想跑得太远，所以她只想沿着通向池塘的最短路径跑步。在牧场里，每条道路连接了两个结点（这些结点的编号为1到N，1≤N≤1000）。另外，如果X>?，说明结点X的地势要高于Y，所以下坡的道路是从X通向Y的，贝西所在牛棚的编号为N（最高点），池塘的编号为1（最低点）。 

而然，一周之后，贝西对单调的路线厌倦了，她希望每天可以跑不同的路线，比如说，最好能有K (1≤K≤100)种不同的选择。为了不至于跑得太累，她希望这Ｋ条路径是从牛棚到池塘的最短的Ｋ条路径。 

请帮助贝西算算她的运动量，即找出网络里最短的K条路径的长度。假设每条道路用(Xi,Yi,Di)表示，其中1≤Yi＜Xi≤N，表示这条道路从Xi出发到Yi，其长度为Di (1≤Di≤1,000,000)。

输入输出格式

输入格式：

• Line 1: Three space-separated integers: N, M, and K

• Lines 2..M+1: Line i+1 describes a downhill cow path using three space-separated integers: X_i, Y_i, and D_i

输出格式：

• Lines 1..K: Line i contains the length of the i-th shortest route or -1 if no such route exists. If a shortest route length occurs multiple times, be sure to list it multiple times in the output.

输入输出样例

输入样例#1：

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 

输出样例#1：

1 
2 
2 
3 
6 
7 
-1 

说明

The routes are (5-1), (5-3-1), (5-2-1), (5-3-2-1), (5-4-3-1),

(5-4-3-2-1).

图论 A* 最短路

突然就忘了重载优先级怎么写

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
const int inf=0x3f3f3f3f;
const int mxn=20010;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{
    int v,nxt,w;
}e[mxn<<1],ve[mxn<<1];
int hd[mxn],mct=0;
int vd[mxn],vct=0;
void add_edge(int u,int v,int w){
    e[++mct].v=v;e[mct].nxt=hd[u];e[mct].w=w;hd[u]=mct;
    ve[++vct].v=u;ve[vct].nxt=vd[v];ve[vct].w=w;vd[v]=vct;
    return;
}
struct node{
    int u,dis;
    node(int a,int b){u=a;dis=b;}
    friend bool operator < (const node &a,const node &b){
        return a.dis>b.dis;
    }
};
priority_queue<node>q;
int dis[mxn];
int n,m,K;
void Dij(){
    memset(dis,0x3f,sizeof dis);
    dis[n]=0;q.push(node(n,0));
    while(!q.empty()){
        node U=q.top();q.pop();
        while(U.dis>dis[U.u]){if(q.empty())break;U=q.top();q.pop();}
        int u=U.u;
//        printf("u:%d
",u);
        for(int i=hd[u];i;i=e[i].nxt){
            int v=e[i].v;
            if(dis[v]>dis[u]+e[i].w){
                dis[v]=dis[u]+e[i].w;
                q.push((node){v,dis[v]});
            }
        }
    }
    return;
}
int tot=0;
struct cmp{
    bool operator () (const node a,const node b){
        return a.dis+dis[a.u]>b.dis+dis[b.u];
    }
};
priority_queue<node,vector<node>,cmp >que;
void solve(){
    que.push(node(1,0));
    while(!que.empty()){
        node U=que.top();que.pop();
        while(U.u==n){
            tot++;
            printf("%d
",U.dis);
            if(tot==K || que.empty())return;
            U=que.top();
            que.pop();
        }
        int u=U.u;
        for(int i=vd[u],v;i;i=ve[i].nxt){
            v=ve[i].v;
            que.push(node(v,U.dis+ve[i].w));
        }
    }
    return;
}
int main(){
    int i,j,u,v,w;
    n=read();m=read();K=read();
    for(i=1;i<=m;i++){
        u=read();v=read();w=read();
        (u>v)?add_edge(u,v,w):add_edge(v,u,w);
    }
    Dij();
    solve();
    for(i=tot+1;i<=K;i++)printf("-1
");
    return 0;
}