L2-001. 緊急救援 SPFA+記錄路徑，統計最短路

作爲一個城市的應急救援隊伍的負責人，你有一張特殊的全國地圖。在地圖上顯示有多個分散的城市和一些連接城市的快速道路。每個城市的救援隊數量和每一條連接兩個城市的快速道路長度都標在地圖上。當其他城市有緊急求助電話給你的時候，你的任務是帶領你的救援隊儘快趕往事發地，同時，一路上召集儘可能多的救援隊。

輸入格式：

輸入第一行給出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<cstdio>
#include<queue>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
const int M =1e4+7;
int n,m,s,d,cnt;
struct edge
{
    int u,v,w;
    int ne;
}e[M*100];
int head[M*100],pre[M],dp[M][2];
int cost[M],dis[M],ans[M];
bool vis[M];
void init(int u,int v,int w)
{
    e[cnt].u=u;
    e[cnt].v=v;
    e[cnt].w=w;
    e[cnt].ne=head[u];
    head[u]=cnt++;
}
void solve()
{
    queue<int>q;
    for(int i=0;i<=n;i++)
    {
        dis[i]=inf;
        vis[i]=0;
    }
    dis[s]=0;
    vis[s]=1;
    q.push(s);
    dp[s][0]=1;
    while(q.size())
    {
        s=q.front();
        q.pop(); vis[s]=0;
        for(int i=head[s];i!=-1;i=e[i].ne)
        {
            int v=e[i].v;
            if(v==s)
                continue;
            if(dis[v]>dis[s]+e[i].w)
            {
                ans[v]=ans[s]+cost[v];
                dis[v]=dis[s]+e[i].w;
                if(!vis[v])
                {
                    q.push(v);
                    vis[v]=1;
                }
                pre[v]=s;
                dp[v][1]=dp[v][0];
                dp[v][0]=dp[s][0];
            }
            else
            if(dis[v]==dis[s]+e[i].w)
            {
               if(!vis[v])
                {
                    q.push(v);
                    vis[v]=1;
                }
                if(ans[v]<ans[s]+cost[v])
                {
                    ans[v]=ans[s]+cost[v];
                    pre[v]=s;
                }
                dp[v][1]=dp[v][0];
                dp[v][0]+=(dp[s][0]-dp[s][1]);
            }
        }
    }
}
void dfs(int x)
{
    if(pre[x]!=-1){
        dfs(pre[x]);
    printf("%d ",pre[x]);
    }
}
int main()
{
    cnt=0;
    memset(ans,0,sizeof(ans));
    memset(dp,0,sizeof(dp));
    memset(pre,-1,sizeof(pre));
    memset(head,-1,sizeof(head));
    cin>>n>>m>>s>>d;
    for(int i=0;i<n;i++){
        cin>>cost[i];
        ans[i]=cost[i];
    }
    while(m--)
    {
        int u,v,w;
        cin>>u>>v>>w;
        init(u,v,w);
        init(v,u,w);
    }
    solve();
    cout<<dp[d][0]<<' '<<ans[d]<<endl;
    dfs(d);
    printf("%d\n",d);
    return 0;
}