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

略微复杂，与以往做的题不同，这里有一个边权和一个点权，而且还要记录路径条数，但距离仍是第一要素，其次是救援队数量，路径和路径条数随之更新，最后递归打印路径。

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>

const int N = 1e3;
const int INF = 0x3f3f3f3f;

int num[N];    //救援队的数量
int map[N][N]; //城市地图
int pre[N];    //存储路径
int n, m, s, d;

//打印路径
void PrintPath(int x)
{
    if(x == -1) return;
    PrintPath(pre[x]);
    printf("%d ", x);  
}

void Dijkstra()
{
    int dis[N];
    bool vis[N];
    int w[N], cnt[N];   //w数组为w[i]的救援队数量，cnt记录最短路径数量
    memset(dis, 0x3f, sizeof(dis));
    memset(vis, 0, sizeof(vis));
    memset(pre, -1, sizeof(pre));  //初始化前驱节点都为-1
    dis[s] = 0;
    w[s] = num[s];
    cnt[s] = 1;
    for(int i = 0; i < n; i++)
    {
        int u = -1, d = INF;
        for(int j = 0; j < n; j++)
        {
            if(!vis[j] && dis[j] < d)
            {
                u = j;
                d = dis[j];
            }
        }
        vis[u] = 1;
        for(int v = 0; v < n; v++)
        {
            if(!vis[v] && map[u][v] != INF)
            {
                if(map[u][v] + dis[u] < dis[v])
                {
                    dis[v] = map[u][v] + dis[u];
                    cnt[v] = cnt[u];
                    pre[v] = u;
                    w[v] = w[u] + num[v];
                }
                else if(map[u][v] + dis[u] == dis[v])
                {
                    cnt[v] += cnt[u];
                    if(w[u] + num[v] > w[v])
                    {
                        w[v] = w[u] + num[v];
                        pre[v] = u;
                    }
                }
            }
        }
    }
    //输出
    printf("%d %d\n", cnt[d], w[d]);
    PrintPath(pre[d]);
    printf("%d", d);
}



int main()
{
    scanf("%d%d%d%d", &n, &m, &s, &d);
    for(int i = 0; i < n; i++)
        scanf("%d", &num[i]);
    memset(map, 0x3f, sizeof(map));
    for(int i = 0; i < m; i++)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        map[a][b] = map[b][a] = c;
    } 
    Dijkstra();
    return 0;
}