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

代码

具体思路看代码注释，主要是理解Dijkstra算法，然后再看清楚题目，已经知道源点和终点，找的是最短路径条数和哪条最短路径的救援队伍最多，dfs试了但是超时了，所以用的Dijkstra算法。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
int g[505][505]; //存图
int a[505],dist[505]; //a存救援队伍，dist源点到此处的距离
int path[505],pa[505],ans[505]; //path存路径，pa是否经过此处，ans存到i点最多的救援队伍
int n,m,s,d,sum[505]; //sum存到i点的路径长度相同的条数
void dij()
{
    for(int i=0;i<n;i++) dist[i]=g[s][i];
    pa[s]=1;
    sum[s]=1; //初始化开始的路径长度相同的条数为1
    dist[s]=0;
    ans[s]=a[s];
    for(int i=1;i<n;i++)
    {
        int temp=INF;
        int v=s;
        for(int j=0;j<n;j++)
        {
            if(!pa[j]&&dist[j]<temp)
            {
                v=j;
                temp=dist[j];
            }
        }
        pa[v]=1;
        for(int j=0;j<n;j++)
        {
            if(!pa[j]&&g[v][j]<INF)
            {
                if(dist[j]>dist[v]+g[v][j]) //找最短路径
                {
                    dist[j]=dist[v]+g[v][j];
                    ans[j]=ans[v]+a[j];  //更新救援队伍
                    path[j]=v;  //点j的前一个点是v
                    sum[j]=sum[v]; //更新路径条数
                }
                else if(dist[j]==dist[v]+g[v][j])
                {
                    if(ans[j]<ans[v]+a[j])
                        ans[j]=ans[v]+a[j],path[j]=v;
                    sum[j]+=sum[v]; //路径长度相等时，说明条数增加，所以要加上前一个点的条数
                }
            }
        }
    }
}
int main()
{
    memset(g,INF,sizeof(g)); //初始化数组
    memset(dist,INF,sizeof(dist));
    cin>>n>>m>>s>>d;
    for(int i=0;i<n;i++) cin>>a[i];
    while(m--)
    {
        int x,y,z;
        cin>>x>>y>>z;
        g[x][y]=z;
        g[y][x]=z;
    }
    for(int i=0;i<n;i++){ //初始化
         if(g[s][i]!=INF) path[i]=s,ans[i]=a[s]+a[i],sum[i]=1;
    }
    dij();
    int j=d;
    stack<int>st; //存路径
    st.push(d);
    while(path[j]!=s) //逆序找路径存入
    {
        st.push(path[j]);
        j=path[j];
    }
    cout<<sum[d]<<" "<<ans[d]<<"\n"; //输出
    cout<<s;
    while(st.size()) 
    {
        cout<<" "<<st.top();
        st.pop();
    }
}