L3-007 天梯地图 (30 分)（两次Dijkstra+DFS求解)

  题目链接：

题目详情 - L3-007 天梯地图 (30 分) (pintia.cn)https://pintia.cn/problem-sets/994805046380707840/problems/994805051153825792

题目描述： 

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

输入格式：

输入在第一行给出两个正整数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

思路：

1. 这题要求找最短时间和最短距离，用两次Dijkstra分别求出时间和距离即可，但是要注意几点

1. 如果最快到达路线不唯一，则输出几条最快路线中最短的那条，题目保证这条路线是唯一的。（意思就是如果最短路径不唯一，则找“最快时间条件下”的最短距离（测试点2））

2. 而如果最短距离的路线不唯一，则输出途径节点数最少的那条，题目保证这条路线是唯一的。（意思就是如果最短距离不唯一，则找经过节点数最少的那条路径）

2. 因为求最快时间时需要“最快时间下的”最短距离，所以先求最快时间的Dijsktra，再求最短距离的Dijkstra（如果先求最短距离，到求最快时间的时候又会被更新，导致最短距离的结果不正确）

3. 最后用DFS分别打印路径即可

代码： 

#include <bits/stdc++.h>
using namespace std;
const int maxn = 505, inf = 0x3f3f3f3f;
// d最短路径，spTime最短时间, Time[a][b]a到b花费的时间
int G[maxn][maxn], d[maxn], vis[maxn], spTime[maxn], Time[maxn][maxn];
int n, m, st, ed, oneWay, MinNum = inf; //st起点到ed终点, oneWay是否单向, MinNum最少节点
// Dispre最短路的前序节点，Timepre最短时间的前序节点，Dispath最短路结果，Timepath最短时间结果
vector<int> Dispre[maxn], Timepre[maxn], tempPath, Dispath, Timepath;

void DijkstraForDis(int s) //最短路的Dijkstra
{
    fill(d, d + maxn, inf);
    d[s] = 0;
    spTime[s] = 0;
    for (int i = 0; i < n; i++) {
        int MIN = inf, u = -1;
        for (int j = 0; j < n; j++)
            if (!vis[j] && d[j] < MIN) {
                MIN = d[j];
                u = j;
            }
        if (u == -1)
            return;
        vis[u] = 1;
        for (int v = 0; v < n; v++) {
            if (!vis[v] && G[u][v] != inf) {
                if (d[u] + G[u][v] < d[v]) { // 找到新的最短路
                    d[v] = d[u] + G[u][v]; // 最短路更新
                    Dispre[v].clear(); // 前序节点更新
                    Dispre[v].push_back(u);
                } else if (d[u] + G[u][v] == d[v]) { // 最短路不唯一
                    Dispre[v].push_back(u); // 添加前序结点
                }
            }
        }
    }
}

void DijkstraForTime(int s) // 最短时间的Dijkstra
{
    fill(spTime, spTime + maxn, inf);
    fill(d, d + maxn, inf); //最短时间不唯一时需要最短路
    d[s] = 0;
    spTime[s] = 0;
    for (int i = 0; i < n; i++) {
        int MIN = inf, u = -1;
        for (int j = 0; j < n; j++)
            if (!vis[j] && spTime[j] < MIN) {
                MIN = spTime[j];
                u = j;
            }
        if (u == -1)
            return;
        vis[u] = 1;
        for (int v = 0; v < n; v++) {
            if (!vis[v] && G[u][v] != inf) {
                if (spTime[u] + Time[u][v] < spTime[v]) { // 新的最短时间
                    spTime[v] = spTime[u] + Time[u][v];
                    Timepre[v].clear();
                    Timepre[v].push_back(u);
                    d[v] = d[u] + G[u][v]; // 更新按最短时间优先的最短路
                } else if (spTime[u] + Time[u][v] == spTime[v]) { // 最短时间不唯一
                    if (d[u] + G[u][v] < d[v]) { // 如最短路更小，则更新
                        Timepre[v].clear(); // 最短路最小唯一，所以前序更新
                        Timepre[v].push_back(u);
                        d[v] = d[u] + G[u][v];
                    }
                }
            }
        }
    }
}

void DisDFS(int u) // 最短路的路径遍历，从终点向起点递归
{
    tempPath.push_back(u); // 存储当前的递归的路径
    if (u == st) { // 到起点
        int num = tempPath.size(); // 最短路节点数
        if (num < MinNum) { // 节点数更少， 更新最短路
            MinNum = num;
            Dispath = tempPath;
        }
        tempPath.pop_back();
        return;
    }
    for (int i = 0; i < Dispre[u].size(); i++) { // 遍历前序节点
        DisDFS(Dispre[u][i]);
    }
    tempPath.pop_back();
}

void TimeDFS(int u) // 最短时间的路径遍历
{
    Timepath.push_back(u); // 保存最短时间的节点
    for (int i = 0; i < Timepre[u].size(); i++) {
        TimeDFS(Timepre[u][i]); // 递归前序节点(其实每个节点前序节点唯一)
    }
}

int main()
{
    // 初始化和输入
    fill(G[0], G[0] + maxn * maxn, inf);
    fill(Time[0], Time[0] + maxn * maxn, inf);
    cin >> n >> m;
    for (int i = 0; i < m; i++) {
        int v1, v2, len, t;
        cin >> v1 >> v2 >> oneWay >> len >> t;
        G[v1][v2] = len;
        Time[v1][v2] = t;
        if (oneWay == 0) { // 如果不是单向路，则双向赋值
            G[v2][v1] = len;
            Time[v2][v1] = t;
        }
    }
    cin >> st >> ed; // 起点，终点
    DijkstraForTime(st); // 先对时间Dijkstra
    memset(vis, 0, sizeof(vis)); // 重新初始化
    DijkstraForDis(st); // 再对距离Dijskra
    DisDFS(ed); // 保存最短距离路径
    TimeDFS(ed); // 保存最短时间路径
    if (Dispath == Timepath) { // 它俩相等
        printf("Time = %d; Distance = %d: ", spTime[ed], d[ed]);
        for (int i = Timepath.size() - 1; i >= 0; i--) {
            printf("%d", Timepath[i]);
            if (i != 0)
                printf(" => ");
        }
    } else { // 不相等
        printf("Time = %d: ", spTime[ed]);
        for (int i = Timepath.size() - 1; i >= 0; i--) {
            printf("%d", Timepath[i]);
            if (i != 0)
                printf(" => ");
        }
        printf("\n");
        printf("Distance = %d: ", d[ed]);
        for (int i = Dispath.size() - 1; i >= 0; i--) {
            printf("%d", Dispath[i]);
            if (i != 0)
                printf(" => ");
        }
    }

    return 0;
}

 总结：

同时考虑最短距离和最快时间，所以两次dijkstra的时候要小心，先求出时间再求出最短距离。