7-47 天梯地图 (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

注释：就是在简单的考Dijkstra算法，但不同的是，在这题中需要运用两次，其实可以先写出其中一个，另一个就直接复制粘贴改下数组名就行了

我刚开始以为路程是距离，结果搞错了，有个测试点没过，在计算t的时候需要开个数组来单独计算路程

#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXV = 520;
struct Node{
    int v, t, d;
    Node(int _v, int _t, int _d) {
        v = _v;
        t = _t;
        d = _d;
    }
};
//这里c是计算距离时节点个数，dt是计算时间的路程
int n, m, d[MAXV], t[MAXV], c[MAXV], preD[MAXV], preT[MAXV], dt[MAXV];
vector<Node> Adj[MAXV];
bool vis[MAXV];
void DijkstraD(int s);
void DijkstraT(int s);
//设立dfs输出过程
void printD(int ed, int cnt);
void printT(int ed, int cnt);
bool compare(int x);

int main(void)
{
    cin >> n >> m;
    int v1, v2, way, time, distance;
    for ( int i = 0; i < m; i ++ ) {
        cin >> v1 >> v2 >> way >> distance >> time;
        Adj[v1].push_back(Node(v2, time, distance));
        if ( way == 0 ) {
            Adj[v2].push_back(Node(v1, time, distance));
        }
    }
    int st, ed;
    cin >> st >> ed;
    DijkstraD(st);
    DijkstraT(st);
    bool flag = compare(ed);
    if ( flag ) {
        cout << "Time = " << t[ed] << "; Distance = " << d[ed] << ": ";
        printD(ed, 0);
    } else {
        cout << "Time = " << t[ed] << ": ";
        printT(ed, 0);
        cout << endl;
        cout << "Distance = " << d[ed] << ": ";
        printD(ed, 0);
        cout << endl;
    }
    
    return 0;
}

void DijkstraD(int s)
{
    fill(d, d + MAXV, INF);
    fill(vis, vis + MAXV, false);
    memset(c, 0, sizeof(c));
    for ( int i = 0; i < MAXV; i ++ ) {
        preD[i] = i;
    }
    d[s] = 0;
    c[s] = 0;
    for ( int i = 0; i < n; i ++ ) {
        int u = -1, MIN = INF;
        for ( int j = 0; j < n; j ++ ) {
            if ( !vis[j] && d[j] < MIN ) {
                MIN = d[j];
                u = j;
            }
        }
        if ( u == -1 ) return ;
        vis[u] = true;
        for ( int j = 0; j < Adj[u].size(); j ++ ) {
            int v = Adj[u][j].v;
            if ( !vis[v] && d[u] + Adj[u][j].d < d[v] ) {
                d[v] = d[u] + Adj[u][j].d;
                c[v] = c[u] + 1;
                preD[v] = u;
            } else if ( !vis[v] && d[u] + Adj[u][j].d == d[v] ) {
                if ( c[u] + 1 < c[v] ) {
                    c[v] = c[u] + 1;
                    preD[v] = u;
                }
            }
        }
    }
}

//直接复制上面的DijkstraD的代码，改下变量名就行了
void DijkstraT(int s)
{
    fill(t, t + MAXV, INF);
    fill(vis, vis + MAXV, false);
    memset(dt, 0, sizeof(dt));
    for ( int i = 0; i < MAXV; i ++ ) {
        preT[i] = i;
    }
    t[s] = 0;
    dt[s] = 0;
    for ( int i = 0; i < n; i ++ ) {
        int u = -1, MIN = INF;
        for ( int j = 0; j < n; j ++ ) {
            if ( !vis[j] && t[j] < MIN ) {
                MIN = t[j];
                u = j;
            }
        }
        if ( u == -1 ) return ;
        vis[u] = true;
        for ( int j = 0; j < Adj[u].size(); j ++ ) {
            int v = Adj[u][j].v;
            if ( !vis[v] && t[u] + Adj[u][j].t < t[v] ) {
                t[v] = t[u] + Adj[u][j].t;
                dt[v] = dt[u] + Adj[u][j].d;
                preT[v] = u;
            } else if ( !vis[v] && t[u] + Adj[u][j].t == t[v] ) {
                if ( dt[u] + Adj[u][j].d < dt[v] ) {
                    dt[v] = dt[u] + Adj[u][j].d;
                    preT[v] = u;
                }
            }
        }
    }
}

//cnt用来控制格式输出
void printD(int ed, int cnt)
{
    if ( preD[ed] != ed ) {
        printD(preD[ed], cnt + 1);
    }
    cout << ed;
    if ( cnt ) cout << " => ";
}

//套用pintD的代码就行了，不需要另外写
void printT(int ed, int cnt)
{
    if ( preT[ed] != ed ) {
        printT(preT[ed], cnt + 1);
    }
    cout << ed;
    if ( cnt ) cout << " => ";
}

//单独拎出来做比较
bool compare(int x)
{
    bool flag = true;
    if ( preD[x] != x || preT[x] != x ) {
        if ( preD[x] == preT[x] ) {
            flag = compare(preD[x]);
        } else {
            flag = false;
        }
    }
    return flag;
}