PAT-001紧急救援

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

题目的关键是求最短路的数量和最短路的解,

我们用前驱数组求最短路径，当当前节点的dis和原先的dis相等时,pathNum[v] += pathNum[u],如果原先的dis较大,就pathNum[v] = pathNum[u];

这点一定要搞清楚

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <map>


using namespace std;
const int inf = 5e2 + 10;
const int maxn = 888888888;
struct node{
int dis;
int v;
}th;
vector<node>s[inf];
int dis[inf];//记录最短距离
int sum[inf];//记录每个城市的救援队的数量
int path[inf];//用它来记录节点路径的前驱
int pathNum[inf];//记录每个节点的从起点到达该点的最短路径数目
int duiNum[inf];//记录到达每个节点的救援队数目
bool fou[inf];
int main()
{
    int n, m;
    int st, ed, st2;
    scanf("%d %d %d %d", &n, &m, &st, &ed);
    st2 = st;
    int ed2 = ed;
    for(int i = 0; i < n; i ++)
        dis[i] = maxn;
    for(int i = 0; i < n; i ++){
        scanf("%d", &sum[i]);
    }
    for(int i = 0; i < m;i ++){
        int u, v;
        scanf("%d %d %d", &u, &v, &th.dis);
        th.v = v;
        s[u].push_back(th);
        th.v = u;
        s[v].push_back(th);
    }
    dis[st] = 0;
    fou[st] = true;
    duiNum[st] = sum[st];
    pathNum[st] = 1;
    int Min;
    int num = 1;
    path[st] = st;
    while(num != n){
        Min = maxn;
        for(int i = 0; i < s[st].size(); i ++){
            if(fou[s[st][i].v]) continue;
            if(dis[s[st][i].v] > dis[st] + s[st][i].dis){
                dis[s[st][i].v] = dis[st] + s[st][i].dis;
                pathNum[s[st][i].v] = pathNum[st];//更新路径数目
                duiNum[s[st][i].v] = duiNum[st] + sum[s[st][i].v];
                path[s[st][i].v] = st;
            }else if(dis[s[st][i].v] == dis[st] + s[st][i].dis){
                pathNum[s[st][i].v] += pathNum[st];
                if(duiNum[s[st][i].v] < duiNum[st] + sum[s[st][i].v]){
                    duiNum[s[st][i].v] = duiNum[st] + sum[s[st][i].v];
                    path[s[st][i].v] = st;
                }
            }
        }
        for(int i = 0; i < n; i ++){
            if(!fou[i] && dis[i] < Min){
                Min = dis[i];
                st = i;
            }
        }
        if(Min == maxn) break;
        fou[st] = true;
        num ++;
    }
    int maxPathNum = 1;
    int way[inf], k = 0;
    cout << pathNum[ed2] << ' ' << duiNum[ed2] << endl;
    while(st2 != ed){
        //maxPathNum = max(maxPathNum, pathNum[ed]);
        way[k ++] = ed;
        ed = path[ed];
    }
    cout << st2;
    for(int i = k - 1; i >= 0; i --) {
        printf(" %d", way[i]);
    }
    cout << endl;
    return 0;
}