L2-001 紧急救援(dijkstra最短路径算法)

1.题目

L2-001 紧急救援
分数 25

全屏浏览

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

输入格式:
输入第一行给出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

2.分析

        本题在dijkstra的算法基础上添加了一些其他条件，题目中给出图的边是无向边，即0可以到1，那么1也可以到0。建议先学习算法原理先写出dijkstra的模板(不计算救援队数量和最短路径的条数)，然后在模板的基础上一步步添加剩余代码。

3.代码

#include<iostream>
#include<vector>
using namespace std;
const int MAX = 500 + 10;
const int INF = 0x3f3f3f;   //定义最小值的初始值

//定义存储图的邻接表，这里是用vector构建
struct edge {
    int from, to, w;
    edge(int a, int b, int c) { from = a; to = b; w = c; } //结构体edge的构造函数
};
vector<edge> e[MAX];

int N, M, S, D, wa[MAX], x, y, z;

//fa[]数组类似并查集，用于存储最短路径结点
int fa[MAX], dist[MAX], uns[MAX], unmber[MAX];//dist[]比较路径，uns[]比较救援队数量，unmber[]记录路径条数
bool done[MAX];   //用于标记结点

void dijkstra() {
    for (int i = 0; i < N; i++) {
        dist[i] = INF;
        done[i] = false;
    }
    dist[S] = 0;
    uns[S] = wa[S];
    unmber[S] = 1;
    for (int i = 0; i < N; i++) {
        int min = INF, minx;
        for (int j = 0; j < N; j++) {
            if (dist[j] < min && !done[j]) {
                min = dist[j];
                minx = j;
            }
        }
        done[minx] = true;
        for (int j = 0; j < e[minx].size(); j++) {
            int t = e[minx][j].to;
            if (min + e[minx][j].w < dist[t] && !done[t]) {
                dist[t] = min + e[minx][j].w;
                uns[t] = uns[minx] + wa[t];
                fa[t] = minx;
                unmber[t] = unmber[minx];
            }
            else if (min + e[minx][j].w == dist[t] && !done[t]) {
                if (uns[minx] + wa[t] > uns[t]) {
                    uns[t] = uns[minx] + wa[t];
                    fa[t] = minx;
                }
                unmber[t] += unmber[minx];
            }
        }
    }
}
int main()
{
    cin >> N >> M >> S >> D;
    for (int i = 0; i < N; i++) {
        cin >> wa[i];
        fa[i] = i;       //初始化
    }
    for (int i = 0; i < M; i++) {
        cin >> x >> y >> z;
        e[x].push_back(edge(x, y, z));
        e[y].push_back(edge(y, x, z));//无向边，两个点都要添加
    }
    dijkstra();
    int p = D, sum = 0;

    //打印最短路径操作
    vector<int> ve;
    while (1) {
        if (p == fa[p]) {
            ve.push_back(p);
            break;
        }
        ve.push_back(p);
        p = fa[p];
    }
    cout << unmber[D] << " " << uns[D] << endl;
    for (int i = ve.size() - 1; i >= 0; i--) {
        if (i == ve.size() - 1) { cout << ve[i]; }
        else { cout << " " << ve[i]; }
    }
    return 0;
}