PTA 7-1 城市间紧急救援

题目详细：

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

输入格式:

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

思路：

和迪杰斯特拉算法一样，都是计算原点到各个点的权值，但是这题有两个优先级不同的权值，一是路径长度，二是救援队数量。你需要在最短路径长度相同时根据救援队数量的多少更新最佳路径。

还有一个问题是最短路径的数量，需要要计算出每一个点的最短路径的数量，假如d是从c来的且是唯一的最短路径，c有两条最短路径，d也就会有2条。如果之后发现经过b到d的路径长度和经过c到d相同，且经b也有2条，那么，d就总共有2+2 = 4条

代码：

#include <bits/stdc++.h>

#define INF 9999999

using namespace std;

typedef struct Path{
    bool hasFinished;
    vector<int> Path;//路径
    int Totallength;//迄今为止的长度
    int TotalRes;//迄今为止的救援队数量
    int NumOfMinPath;//最短路数量
} Path;//数组下标是它对应的点

int source,dest,N,M;
int ** Map;
int * NumOfResOfEachCity;
Path * info;

void GetMap();
void FindMinPath();
void Print();

int main()
{
    GetMap();
    FindMinPath();
    Print();
}

void GetMap()
{
    scanf("%d%d%d%d",&N,&M,&source,&dest);
    NumOfResOfEachCity = (int*)malloc(sizeof(int)*N);
    for(int i = 0;i<N;i++)
        scanf("%d",&NumOfResOfEachCity[i]);
    Map = (int **)malloc(sizeof(int *)*N);
    for(int i = 0;i<N;i++)
        Map[i] = (int *)malloc(sizeof(int)*N);
    for(int i = 0;i<N;i++)
        for(int j = 0;j<N;j++)
            Map[i][j] = INF;
    for(int j = 0;j<M;j++)
    {
        int x,y,length;
        scanf("%d%d%d",&x,&y,&length);
        Map[x][y] = length;
        Map[y][x] = length;
    }
}
void FindMinPath()//迪杰斯特拉算法
{
    //开辟信息表格
    info = new Path[N];
    //初始化源到dest点的距离等各种信息
    for(int i = 0;i<N;i++)
    {
        info[i].NumOfMinPath = 1;//原点到每个点至少有一条路
        info[i].Totallength = Map[source][i];
        info[i].hasFinished = false;
        if( info[i].Totallength != INF )
        {
            info[i].Path.push_back(source);
            info[i].Path.push_back(i);
        }
        if( i!= source)
            info[i].TotalRes = NumOfResOfEachCity[source]+NumOfResOfEachCity[i];
    }
    //初始化原点
    info[source].hasFinished = true;
    info[source].TotalRes = NumOfResOfEachCity[source];
    //循环n-1次，每次找最小值，并根据新加入的点更新
    for(int i = 1;i < N-1;i++)
    {
        int min = INF;
        int flag;
        for(int j = 0;j<N;j++)
        {
            if(info[j].hasFinished == false&&info[j].Totallength<min)
            {
                min = info[j].Totallength;
                flag = j;
            }
        }
        info[flag].hasFinished = true;
        for(int j = 0;j<N;j++)
        {
            if(info[j].hasFinished == false&&Map[flag][j]+info[flag].Totallength<info[j].Totallength)
            { 
                info[j].Totallength = Map[flag][j]+info[flag].Totallength;
                info[j].TotalRes = NumOfResOfEachCity[j]+info[flag].TotalRes;
                info[j].NumOfMinPath = info[flag].NumOfMinPath;//此点会继承上一个点的路径数
                //更新路径
                info[j].Path.assign(info[flag].Path.begin(), info[flag].Path.end());
                info[j].Path.push_back(j);
            }
            else if(info[j].hasFinished == false&&Map[flag][j]+info[flag].Totallength==info[j].Totallength)
            {
                info[j].NumOfMinPath+=info[flag].NumOfMinPath;//加上从另一个路径长度相同的点来的路径数
                if(info[flag].TotalRes+NumOfResOfEachCity[j]>info[j].TotalRes)
                {
                    info[j].TotalRes = info[flag].TotalRes+NumOfResOfEachCity[j];
                    info[j].Path.assign(info[flag].Path.begin(), info[flag].Path.end());
                    info[j].Path.push_back(j);
                }
            }
        }
    }
}
void Print()
{
    printf("%d %d\n",info[dest].NumOfMinPath,info[dest].TotalRes);
    for(int i = 0;i<info[dest].Path.size();i++)
    {
        if(i==0)
            printf("%d",info[dest].Path[i]);
        else
            printf(" %d",info[dest].Path[i]);
    }
}