PAT L2-001. 紧急救援 (最短路变形 Dijkstra + 记录路径)

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

输入格式：

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

最开始看见这道题觉得无从下手，主要是不会记录最短路径的个数，最后通过学习知道了要得到这一点的路径个数有两种方面的更新

1> 跟随着路径的更新，把step更新

if(dis[i] > dis[x]+e[x][i])
            {
                A[i] = A[x]+army[i];
                dis[i] = dis[x]+e[x][i];
                rode[i] = x;
                if(step[i] < step[x]) step[i] = step[x];
            }

2> 路径相等的情况下，应加上另一条边的路径数

else if(dis[i] == dis[x]+e[x][i])
            {
                if(A[i] < A[x]+army[i])
                {
                    A[i] = A[x]+army[i];
                    rode[i] = x;
                }
                step[i] += step[x];
            }

A[i] 和 dis[i] 分别代表救援队数量和最短路径，其计算方式类似

#include<stdio.h>
#include<string.h>
#define INF 0x3f3f3f3f

int e[510][510],army[510];//每个点的救援队

int n,m,s,d;

void Rode()
{
    int step[510] = {0},rode[510] = {0};//存储最短路的个数与路径
    int dis[510],A[510] = {0};//dis~i点到源点的最短距离，A~i点到源点的最多救援队
    for(int i = 0; i<n; i++)
    {
        dis[i] = e[i][s];
        if(dis[i]<INF)//与i点相连的点
        {
            step[i] = 1;
            A[i] = army[i]+army[s];
            rode[i] = s;
        }
    }
    dis[s] = 0;
    int book[510] = {0},k = 1;
    book[s] = 1;
    
    while(1)
    {
        int Min = INF,x = 0;
        for(int i = 0; i<n; i++)
            if(book[i] == 0 && dis[i] < Min)
            {
                Min = dis[i];
                x = i;
            }
        if(x == d) break;
        book[x] = 1;
        for(int i = 0; i<n; i++)
        {
            if(dis[i] > dis[x]+e[x][i])
            {
                A[i] = A[x]+army[i];
                dis[i] = dis[x]+e[x][i];
                rode[i] = x;
                if(step[i] < step[x]) step[i] = step[x];
            }
            else if(dis[i] == dis[x]+e[x][i])
            {
                if(A[i] < A[x]+army[i])
                {
                    A[i] = A[x]+army[i];
                    rode[i] = x;
                }
                step[i] += step[x];
            }
        }
    }
    printf("%d %d\n",step[d],A[d]);
    int R[510],i = d,j = 0;
    while(i!=s)
    {
        R[j++] = i;
        i = rode[i];
    }
    printf("%d",s);
    for(j--; j>=0; j--)
        printf(" %d",R[j]);
    printf("\n");
}
int main()
{

    while(~scanf("%d%d%d%d",&n,&m,&s,&d))
    {
        memset(e,INF,sizeof(e));
        for(int i = 0; i<n; i++)
        {
            scanf("%d",&army[i]);
        }
        int a,b,c;
        for(int i = 0; i<m; i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            if(e[a][b] < c) continue;
            e[a][b] = e[b][a] = c;
        }
        Rode();
    }
}