7-4 天梯地图 （30 分）

7-4 天梯地图 （30 分）

本题要求你实现一个天梯赛专属在线地图，队员输入自己学校所在地和赛场地点后，该地图应该推荐两条路线：一条是最快到达路线；一条是最短距离的路线。题目保证对任意的查询请求，地图上都至少存在一条可达路线。
输入格式：

输入在第一行给出两个正整数N（2 ≤ N ≤ 500）和M，分别为地图中所有标记地点的个数和连接地点的道路条数。随后M行，每行按如下格式给出一条道路的信息：

V1 V2 one-way length time

其中V1和V2是道路的两个端点的编号（从0到N-1）；如果该道路是从V1到V2的单行线，则one-way为1，否则为0；length是道路的长度；time是通过该路所需要的时间。最后给出一对起点和终点的编号。
输出格式：

首先按下列格式输出最快到达的时间T和用节点编号表示的路线：

Time = T: 起点 => 节点1 => … => 终点

然后在下一行按下列格式输出最短距离D和用节点编号表示的路线：

Distance = D: 起点 => 节点1 => … => 终点

如果最快到达路线不唯一，则输出几条最快路线中最短的那条，题目保证这条路线是唯一的。而如果最短距离的路线不唯一，则输出途径节点数最少的那条，题目保证这条路线是唯一的。

如果这两条路线是完全一样的，则按下列格式输出：

Time = T; Distance = D: 起点 => 节点1 => … => 终点

输入样例1：

10 15
0 1 0 1 1
8 0 0 1 1
4 8 1 1 1
5 4 0 2 3
5 9 1 1 4
0 6 0 1 1
7 3 1 1 2
8 3 1 1 2
2 5 0 2 2
2 1 1 1 1
1 5 0 1 3
1 4 0 1 1
9 7 1 1 3
3 1 0 2 5
6 3 1 2 1
5 3

输出样例1：

Time = 6: 5 => 4 => 8 => 3
Distance = 3: 5 => 1 => 3

输入样例2：

7 9
0 4 1 1 1
1 6 1 3 1
2 6 1 1 1
2 5 1 2 2
3 0 0 1 1
3 1 1 3 1
3 2 1 2 1
4 5 0 2 2
6 5 1 2 1
3 5

输出样例2：

Time = 3; Distance = 4: 3 => 2 => 5

#include <bits/stdc++.h>
#define to first
#define dist second.first
#define time second.second
using namespace std;
const int maxn = 520;
const int inf = 0x3f3f3f3f;
typedef pair<int, int> pii;
vector<pair<int, pii> > E[maxn];
int n,m,S,D;
int pre_d[maxn], pre_t[maxn], cnt[maxn], _d[maxn], d[maxn];
bool vis[maxn];
void Dijkstra(int flag)
{
    memset(vis, false, sizeof vis);
    memset(cnt, 0, sizeof cnt);
    priority_queue<pii, vector<pii>, greater<pii> > pq;
    pq.push({0, S});
    if(flag) {
        memset(pre_t, -1, sizeof pre_t);
        memset(_d, 0x3f, sizeof _d);
        cnt[S] = 1;
        _d[S] = 0;
        pii tmp;
        while(!pq.empty()) {
            tmp = pq.top(); pq.pop();
            int u = tmp.second;
            if(vis[u]) continue;
            vis[u] = true;
            for(int i = 0; i < E[u].size(); i++) {
                int v = E[u][i].to;
                int w = E[u][i].time;
                int dd = E[u][i].dist;
                if(vis[v]) continue;
                if(_d[v] > _d[u] + w) {
                    _d[v] = _d[u] + w;
                    pq.push({_d[v], v});
                    pre_t[v] = u;
                    cnt[v] = cnt[u] + dd;
                }
                else if(_d[v] == _d[u] + w) {
                    if(cnt[v] > cnt[u] + dd) {
                        cnt[v] = cnt[u] + dd;
                        pre_t[v] = u;
                    }
                }
            }
        }
    }
    else {
        memset(pre_d, -1, sizeof pre_d);
        memset(d, 0x3f, sizeof d);
        pii tmp;
        d[S] = 0;
        while(!pq.empty()) {
            tmp = pq.top(); pq.pop();
            int u = tmp.second;
            if(vis[u]) continue;
            vis[u] = true;
            for(int i = 0; i < E[u].size(); i++) {
                int v = E[u][i].to;
                int w = E[u][i].dist;
                if(vis[v]) continue;
                if(d[v] > d[u] + w) {
                    d[v] = d[u] + w;
                    pq.push({d[v], v});
                    pre_d[v] = u;
                    cnt[v] = cnt[u] + 1;
                }
                else if(d[v] == d[u] + w) {
                    if(cnt[v] > cnt[u] + 1) {
                        cnt[v] = cnt[u] + 1;
                        pre_d[v] = u;
                    }
                }
            }
        }
    }
}
void print_ans(int pre[], int p) {
    if(pre[p] == -1) {
        cout << p;
    }
    else {
        print_ans(pre, pre[p]);
        cout << " => " << p;
    }
}

bool _same()
{
    int t = D;
    while(t != -1) {
        if(pre_t[t] != pre_d[t]) return 0;
        t = pre_d[t];
    }
    return 1;
}

int main()
{
    //freopen("i.txt", "r", stdin);
    cin >> n >> m;
    for(int k = 0, u, v, isone, dis, ti; k < m; k++) {
        cin >> u >> v >> isone >> dis >> ti;
        E[u].push_back({v, {dis, ti}});
        if(!isone) {
            E[v].push_back({u, {dis, ti}});
        }
    }
    cin >> S >> D;
    Dijkstra(1);
    Dijkstra(0);
    if(_same()) {
        printf("Time = %d; Distance = %d: ", _d[D], d[D]);
        print_ans(pre_d, D);
    }
    else
    {
        printf("Time = %d: ", _d[D]);
        print_ans(pre_t, D);
        cout << endl;
        printf("Distance = %d: ", d[D]);
        print_ans(pre_d, D);
        cout << endl;
    }
    return 0;
}