PTA团体程序设计天梯赛-练习集 L2-001 紧急救援(Dijkstra)

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。
输入格式：
输入第一行给出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
思路：用一遍Dijkstra算法，weight数组记录能召集的最大救援队，cnt数组记录最短路径的条目。pre数组记录最优解路径。

#include <bits/stdc++.h>
using namespace std;
const int inf = 99999999;
int n,m,s,d,num[505],e[505][505],dis[505],weight[505],vis[505],pre[505],cnt[505];
void printPath(int v) {
    if(v == s) {
        printf("%d", v);
        return ;
    }
    printPath(pre[v]);
    printf(" %d", v);
}
int main() {
    cin>>n>>m>>s>>d;
    for(int i = 0;i < n;i++){
        cin>>num[i];
    }
    fill(e[0],e[0]+505*505,inf);
    fill(dis, dis + 505, inf);
    for(int i = 0;i < m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        e[a][b] = c;
        e[b][a] = c;
    }
    dis[s] = 0;
    weight[s] = num[s];
    cnt[s] = 1;
    for(int i = 0;i < n;i++){
        int flag = -1,minn = inf;
        for(int j = 0;j < n;j++){
            if(vis[j] == 0 && dis[j] < minn){
                flag = j;
                minn = dis[j];
            }
        }
        if(flag == -1) break;
        vis[flag] = 1;
        for(int j = 0;j < n;j++){
            if(vis[j] == 0 && e[flag][j] != inf) {
                if (dis[flag] + e[flag][j] < dis[j]) {
                    weight[j] = weight[flag] + num[j];
                    pre[j] = flag;
                    cnt[j] = cnt[flag];
                    dis[j] = dis[flag] + e[flag][j];
                }else if(dis[flag] + e[flag][j] == dis[j]){
                    cnt[j] = cnt[j] + cnt[flag];
                    if(weight[flag] + num[j] > weight[j]) {
                        weight[j] = weight[flag] + num[j];
                        pre[j] = flag;
                    }
                }
            }
        }
    }
    cout<<cnt[d]<<" "<<weight[d]<<endl;
    printPath(d);
    return 0;
}