城市间紧急救援(Dijkstra+输出路径)

题目链接：L2-001 紧急救援 （25 分)

题意：n个城市，每个城市有人数，给你起点和终点，求最短路，如果有多条，输出最短路径条数和经过的城市人数之和最大的一条。

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;

const int N = 510;
int vis[N];//是否用这个点更新过
int ma[N][N];//存图
int dis[N];//两点间最短距离
int num[N];//每个点的救援队人数
int numdis[N];//到每个点最多能够集结到的人数
int path[N];//存路径
int pathnum[N];//到达每个点的路径数量
int n,m,s,d;
int ans[N],top = 0;//用于输出路径

void dj()
{
    memset(vis,0,sizeof(vis));//初始化
    memset(dis,inf,sizeof(dis));//一定要记得初始化
    pathnum[s] = 1;//起点肯定有一条路
    numdis[s] = num[s];//起点的救援队
    path[s] = -1;
    dis[s] = 0;
    for(int i = 1; i <= n; i++)
    {
        int mn = inf,u;
        for(int j = 0; j < n; j++)//板子
        {
            if(!vis[j] && dis[j] < mn)
            {
                mn = dis[j];
                u = j;
            }
        }
        vis[u] = 1;
        for(int j = 0; j < n; j++)
        {
            if(!vis[j])
            {
                if(dis[j] > dis[u]+ma[u][j])//能够更新最短路的时候必须更新
                {
                    dis[j] = dis[u]+ma[u][j];
                    path[j] = u;//记录路径
                    //cout << path[j] << endl;
                    pathnum[j] = pathnum[u];//路径数不变
                    numdis[j] = numdis[u]+num[j];//救援队数量相加
                }
                else if(dis[j] == dis[u]+ma[u][j])//如果距离相等
                {
                    pathnum[j] += pathnum[u];//首先肯定路径条数增加
                    if(numdis[j] < numdis[u]+num[j])//如果人数能变多就更新
                    {
                        numdis[j] = numdis[u]+num[j];//更新到该点的人数
                        path[j] = u;//记录路径
                        //cout << path[j] << endl;
                    }
                }
            }
        }
    }
}

void getpath(int x)
{
    if(path[x] != -1)
    {
        ans[top++] = x;
        getpath(path[x]);
    }
    else
    {
        ans[top++] = x;
        return ;
    }
}

int main()
{
    int u,v,w;
    cin >> n >> m >> s >> d;
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            ma[i][j] = inf;
        }
        ma[i][i] = 0;
    }
    for(int i = 0; i < n; i++)
    {
        cin >> num[i];
    }
    for(int i = 1; i <= m; i++)
    {
        cin >> u >> v >> w;
        ma[u][v] = w;
        ma[v][u] = w;
    }
    dj();
    getpath(d);
    cout << pathnum[d] << ' ' << numdis[d] << endl;
    for(int i = top-1; i > 0; i--)
    {
        cout << ans[i] << ' ';
    }
    cout << ans[0] << endl;
    return 0;
}