7-1 城市间紧急救援 （25 分）第七章-图-最短路-计算机17级

7-1 城市间紧急救援 （25 分）

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

输入格式:

输入第一行给出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 <iostream>
#include <bits/stdc++.h>
#define maxn 501
using namespace std;
const int INF = 0x3f3f3f3f;
int mp[maxn][maxn];//邻接矩阵mp
int path[maxn],dist[maxn];
int N,M,S,D;//N是城市的个数，顺便假设城市的编号为0 ~ (N−1)；
//M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。
int num[maxn];//第i个城市的救援队的数目
int cnt[maxn],pnum[maxn];//cnt统计最短路径条数，pnum统计最多的救援队数量

/*
int FindMinDist(int dist[], int collected[])
{
    int MinDist = INF;
    int S,MinV;
    MinV = S;
    for(int V = 0;V < N;V++)
    {
        if(collected[V] == false && dist[V] < MinDist)
        {
            MinDist = dist[V];
            MinV = V;
        }
    }
    if (MinDist < INF)  若找到最小dist
    {
        collected[MinV] = true;
        return MinV; /返回对应的顶点下标
    }
    else
        return ERROR;  若这样的顶点不存在，返回错误标记
}
*/

void Dijkstra()
{
    int collected[maxn];//判断是否收录
    memset(cnt,0,sizeof cnt);
    memset(collected,false,sizeof collected);
    memset(dist,INF,sizeof dist);
    memset(path,-1,sizeof path);
    memset(pnum,0,sizeof pnum);
    /*
    for(int V = 0;V < N;V++)//初始化
    {
        dist[V] = mp[S][V];
        if(dist[V] < INF)
            path[V] = S;
        else
            path[V] = -1;
    }
    */
    dist[S] = 0;//自己到自己的距离为0
    collected[S] = true;
    cnt[S] = 1, pnum[S] = num[S];
    for(int i = 1;i < N;i++)
    {
        int MinDist = INF;
        int MinV = S ;
        for(int V = 0; V < N; V++)
        {
            if(collected[V] == false && dist[V] < MinDist)
            {
                MinDist = dist[V];
                MinV  = V;
            }
        }
        collected[MinV] = true;
        //V = FindMinDist(dist[i],collected[i]);
        for(int W = 0;W < N;W++)
        {
            if(collected[W] == false)
            {
                if(dist[MinV] + mp[MinV][W] < dist[W])
                {
                    dist[W] = dist[MinV] + mp[MinV][W];//更新dist
                    path[W] = MinV;//记录路径
                    cnt[W] = cnt[MinV];
                    pnum[W] = pnum[MinV] + num[W];
                }
                else if(dist[MinV] + mp[MinV][W] == dist[W])
                {
                    cnt[W] = cnt[MinV] + cnt[W];
                    if(pnum[W] < pnum[MinV] + num[W])
                    {
                        pnum[W] = pnum[MinV] + num[W];
                        path[W] = MinV;
                    }
                }
            }
        }
    }
}


void print()
{
    stack<int> s;
    int j = D;//j是目的地的编号
    while(path[j] != -1)
    {
        s.push(j);
        j = path[j];//更新j为他上一个顶点
    }
    s.push(j);
    cout<<s.top();
    s.pop();
    while(!s.empty())
    {
        cout<<' '<<s.top();
        s.pop();
    }
}

int main()
{
    memset(mp,INF,sizeof mp);
    for(int i = 0; i < maxn ; i++)
        mp[i][i]  = 0;
    int st,ed,cost;
    cin>>N>>M>>S>>D;
    for(int i = 0; i < N; i++)
        cin>>num[i];
    for(int i = 0; i< M ; i++)
    {
        cin>>st>>ed>>cost;
        mp[st][ed] = mp[ed][st] = cost;//初始化邻接矩阵（双向图）
    }
    Dijkstra();
    cout<<cnt[D]<<' '<<pnum[D] <<endl;
    print();
    return 0;
}