城市间紧急救援 (邻接表c++)

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

输入格式:

输入第一行给出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<bits/stdc++.h>
using namespace std;
#define inf (int)1e9 + 1
#define MAXN 520

typedef struct Node{
    int v;
    int dis;
    Node(int _v , int _dis) : v(_v) , dis(_dis){}
}node;

int people[MAXN];
int p[MAXN];
int d[MAXN];
bool vis[MAXN];
int num[MAXN];
int pre[MAXN]; //记录路径前驱
int N , M , S , D;

void Dijkstra(int s , vector<node> Adj[]);
void dfs(int v);

int main(){
    cin >> N >> M >> S >> D;
    vector<node> Adj[N + 1];
    for(int i = 0;i < N;i ++){
        cin >> people[i];
    }
    for(int i = 0;i < M;i ++){
        int v1 , v2 , dis;
        cin >> v1 >> v2 >> dis;
        Adj[v1].emplace_back(Node(v2 , dis));
        Adj[v2].emplace_back(Node(v1 , dis));
    }
    Dijkstra(S , Adj);
    cout << num[D] << " " << p[D] << endl;
    dfs(D);
    return 0;
}

void Dijkstra(int s , vector<node> Adj[]){
    fill(d , d + MAXN , inf);
    memset(num , 0 , sizeof(int) * (N + 1));
    memset(p , 0 , sizeof(int) * (N + 1));
    d[s] = 0;//起点s到自身的距离为零
    p[s] = people[s];//起点的救援队数量为到达起点的最多救援队数量
    num[s] = 1; //第一次只有起点s到自身有一条最短路径
    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){//找到未访问顶点中到起点距离最小的
                u = j;
                min = d[j];
            }
        }
        //i == 0时，即第一次找的点永远是起点
        if(u == -1){
            return;
        }
        vis[u] = 1; //标记u已访问
        for(int j = 0;j < Adj[u].size();j ++){ //获得u能到达的点v
            int v = Adj[u][j].v;
            if(!vis[v] && d[u] + Adj[u][j].dis < d[v]){//v未访问且u作为中介点能使s到达v的距离缩短
                d[v] = d[u] + Adj[u][j].dis; //覆盖
                p[v] = p[u] + people[v];
                num[v] = num[u]; //从s到达当前节点的最短路径数量等于从s到达上一节点的最短路径数量
                pre[v] = u;
            }else if(!vis[v] && d[u] + Adj[u][j].dis == d[v]){
                if(p[u] + people[v] > p[v]){ //在距离一致的条件下u作为中介点能使s到达v的救援队数量增加
                    p[v] = p[u] + people[v];
                    pre[v] = u;
                }
                num[v] += num[u]; //从s到达v的最短路径增加了（从s到u的最短路径的数量）条         
            }
        }
    }
}

void dfs(int v){
    if(v == S){
        cout << v;
        return;
    }else{
        dfs(pre[v]);
        cout << " " << v;
    }
}

将num[v] ++ 改成 num[v] += num[u]全部正确