【GPLT 二阶题目集】L2-001 城市间紧急救援

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式：

输入第一行给出4个正整数N、M、S、D，其中N（2≤N≤500）是城市的个数，顺便假设城市的编号为0 ~ (N−1)；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。

第二行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式：

第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出结尾不能有多余空格。

输入样例：

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

输出样例：

2 60
0 1 3

#include <iostream>
#include <algorithm>
#include <string.h> //memset函数，本来用来初始化，后发现根据字节只能初始为0，避坑
#include <map>
#include <vector>
#define CITY_MAXNUM 500
#define MAXSIZE 999
using namespace std;

int find_min(int temp[], int n)
{
    int min_value = temp[0], min_num = 0;
    for (int i = 1; i < n; i++) {
        if (temp[i] < min_value) {
            min_value = temp[i]; min_num = i;
        }
    }
    return min_num;
}

int main()
{
    int n, m, s, d; cin >> n >> m >> s >> d;
    int rescue[CITY_MAXNUM]; //各城市救援队数量
    int Map[CITY_MAXNUM][CITY_MAXNUM]; //城市间快速道路长度
    int count[CITY_MAXNUM]; //重复的最短路线数量
    for (int i = 0; i < n; i++)
        count[i] = 1;
    map< int, vector<int> > way; //路线
    for (int i = 0; i < n; i++) //地图道路初始化
        for (int j = 0; j < n; j++)
            Map[i][j] = MAXSIZE;
    int temp, a, b;
    vector<int> v; v.push_back(s);
    for (int i = 0; i < n; i++) { //城市救援队数量
        cin >> temp; rescue[i] = temp;
        way.insert(make_pair(i, v));
    }
    for (int i = 0; i < m; i++) { //两城市快速道路长度
        cin >> a >> b >> temp;
        Map[a][b] = Map[b][a] = temp;
    }

    int temp2[CITY_MAXNUM]; //当前情况下s到各城市的最短路径
    int temp3[CITY_MAXNUM]; //当前情况下s到各城市时获得的救援队数量
    for (int i = 0; i < n; i++)
        temp2[i] = MAXSIZE + 100;
    memset(temp3, 0, n * sizeof(int));
    int min_len[CITY_MAXNUM]; //s到该城市的最短路径
    bool isMin[CITY_MAXNUM] = { false }; //是否已经找到s到该城市的最短路径

    min_len[s] = 0, isMin[s] = true, temp3[s] = rescue[s];
    int start = s;//初始出发点
    for (int i = 0; i < n - 1; i++) {
        for (int j = 0; j < n; j++) {
            if (!isMin[j]) { //s到j的最短路径=min(上一次最短路径，s经过start到j)
                if (min_len[start] + Map[start][j] < temp2[j]) {
                    temp2[j] = min_len[start] + Map[start][j];
                    temp3[j] = rescue[j] + temp3[start];
                    way[j] = way[start]; way[j].push_back(j);
                    count[j] = count[start];
                }
                else if (min_len[start] + Map[start][j] == temp2[j]) {
                    //count[j]++;
                    count[j] += count[start];
                    if (temp3[j] < rescue[j] + temp3[start]) {
                        temp3[j] = rescue[j] + temp3[start];
                        way[j] = way[start]; way[j].push_back(j);
                    }
                }
            }
            else temp2[j] = MAXSIZE;
        }
        start = find_min(temp2, n); //新的出发点
        min_len[start] = temp2[start], isMin[start] = true;//记录s到该城市的最短路径
        if (start == d) break;//抵达目的地，退出
    }
    /*int res = 1;
    for (int i = 0; i < way[d].size(); i++) {
        res *= (++count[way[d][i]]);
    }*/
    cout << count[d] << " " << temp3[d] << endl;
    for (int i = 0; i < way[d].size(); i++) {
        cout << way[d][i];
        if (i != way[d].size()-1) cout << " ";
        else cout << endl;
    }
    return 0;
}

 注意事项：

这道题目非常折磨人，尤其测试点1，我用累乘计算最短路径条数，各种自测用例能过，但是测试点1一直答案错误，只能改成继承/累加来算最短路径条数。

尽管如此，不知道之前的方法问题出在哪，希望有能人解答，万分感谢！

如有问题，欢迎提出。