团体程序设计天梯赛-练习集L2-001. 紧急救援（dijkstra）

L2-001. 紧急救援

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出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算法的加强版

#include<bits/stdc++.h>
#define inf 1e9+7
using namespace std;
stack<int >q;
struct node {
    long long value = -1;//表示从起点到这点的总救援队数目，总价值
    int cost = inf;//表示从这点出发的最短长度
    int pointvalue;//表示这点拥有的救援队数目
    int vis = 0;//表示这点是否访问过
    int cnt = 0;//表示这点拥有的最短路径条数
    int fa;//寻找起点
};
node arr[505];
int mat[505][505];

void dijkstra(int start, int n) {
    int now = start;
    arr[start].cnt = 1;
    arr[start].vis = 1;
    arr[start].value = arr[start].pointvalue;
    arr[start].cost = 0;
    arr[start].fa = start;
    for(int i = 0; i < n; i++) {
        int minn = inf;
        for(int j = 0; j < n; j++) { //统计起点到每个点的距离
            if(now == j) continue;
            if(arr[now].cost != inf && arr[now].cost + mat[now][j] < arr[j].cost && arr[j].vis == 0) {//找到这点到与之联通的点的最短距离
                arr[j].cost = arr[now].cost + mat[now][j];
                arr[j].value = arr[now].value + arr[j].pointvalue;
                arr[j].fa = now;
                arr[j].cnt = arr[now].cnt;
            } else if(arr[now].cost + mat[now][j] == arr[j].cost) { //相等长度则把救援队的数量相加
                arr[j].cnt += arr[now].cnt;
                if(arr[j].value < arr[now].value + arr[j].pointvalue) {
                    arr[j].value = arr[now].value + arr[j].pointvalue;
                    arr[j].fa = now;
                }
            }
        }
        for(int j = 0; j < n; j++) { //找到这点到与之联通的最短长度的点后则到下一个点
            if(now == j) continue;
            if(arr[j].vis == 0 && arr[j].cost < minn) {
                minn = arr[j].cost;
                now = j;
            }
        }
        arr[now].vis = 1;
    }
}
int main() {
    int n, m, start, end;
    cin >> n >> m >> start >> end;
    for(int i = 0; i < n; i++)
        cin >> arr[i].pointvalue;//救援队数目(点权);
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            mat[i][j] = inf;
        }
    }
    for(int i = 0; i < m; i++) {
        int s, e, k;
        cin >> s >> e >> k;
        mat[s][e] = mat[e][s] = k;
    }
    dijkstra(start, n);
    cout << arr[end].cnt << ' ' << arr[end].value << endl;
    while(arr[end].fa != end) {
        q.push(end);
        end = arr[end].fa;
    }
    q.push(start);
    while(!q.empty()) {
        cout << q.top();
        q.pop();
        if(!q.empty())
            cout << ' ';
    }
    cout << endl;
    return 0;
}