PTA程序设计天梯赛L2-001 紧急救援

题目

PTA程序设计天梯赛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
原题链接：L2-001紧急救援

题目大意

给定了n个城市，m条边，每个城市有一个对应的救援队数量，求最短路径的条数和经过这些点对应的救援队最大和，输出路径。

预备知识

最短路径算法，博主使用的是堆优化dijkstra，最短路径的存储。

思路和算法

最短路径可以通过dijkstra算法求出，但是该题要求最短路径条数、最多救援队数量以及整个路径。

1. 最短路径条数
   dijkstra算法中需要一个距离数组dis，表示起点到其余每个点的最短距离
   我们可以在开辟一个path数组，表示起点到该点最短路径的条数，在更新最短距离时更新path数组：
   如果i点距离可以更小，那么path[i] = 1，如果从j点找到了和i点距离一样的其他路，那么path[i] += path[j]
2. 最多救援队数量
   类似的，开辟tol数组表示到i点经过最短路径能最多召集的救援队数量
3. 路径输出
   开辟pre数组，pre[i] = j表示从起点到i的最短路径中，i的前一个点是j

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 510, M = 510 * 510;
int h[N], e[M], ne[M], w[M], idx, help[N];
int n, m, s, d;
int pre[N], path[N], tol[N], dis[N];
bool st[N];
typedef pair<int, int> pii;

void add(int a, int b, int c) {//链式前向星存图
    e[idx] = b;
    ne[idx] = h[a];
    w[idx] = c;
    h[a] = idx++;
}

void dijkstra() {
    priority_queue<pii, vector<pii>, greater<pii>> heap;
    heap.push({0, s});
    dis[s] = 0;
    path[s] = 1;
    pre[s] = -1;
    tol[s] = help[s];
    while(heap.size()) {
        pii t = heap.top();
        heap.pop();
        int v = t.second, dist = t.first;
        if(st[v]) continue;
        st[v] = 1;
        if(v == d) return;
        for(int i = h[v]; i != -1; i = ne[i]) {
            int j = e[i];
            if(dist + w[i] < dis[j]) {//如果找到了更短的路径，就重新更新
                dis[j] = dist + w[i];
                pre[j] = v;
                path[j] = path[v];
                tol[j] = tol[v] + help[j];
                heap.push({dis[j], j});
            }else if(dist + w[i] == dis[j]) {// 如果找到了和之前相等的路径，就加上
                path[j] += path[v];
                if(tol[j] < tol[v] + help[j]) {
                    tol[j] = tol[v] + help[j];
                    pre[j] = v;
                }
            }
        }
    }
}

void print() {//已知终点，使用栈逆序输出就能得到起点到终点
    stack<int> sk;
    int x = d;
    while(x != -1) {
        sk.push(x);
        x = pre[x];
    }
    while(sk.size()) {
        cout << sk.top();
        sk.pop();
        if(sk.size()) cout << " ";
        else cout << endl;
    }
}

int main() {
    cin >> n >> m >> s >> d;
    for(int i = 0; i < n; i++) cin >> help[i];
    memset(dis, 0x3f, sizeof(dis));
    memset(h, -1, sizeof(h));
    while(m--) {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);//注意是无向图
        add(b, a, c);
    }
    dijkstra();
    cout << path[d] << " " << tol[d] << endl;
    print();
    return 0;
}

复杂度分析

时间复杂度

使用堆优化dijkstra，时间复杂度为O(mlogn)

空间复杂度

主要用来存边，空间复杂度为O(m)