维护最短路径条数和途径点的权值累加

 团体程序设计天梯赛L2-001紧急救援(25分)

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstring>
#define x first
#define y second
using namespace std;
typedef pair<int,int>PII;
const int N=510;
priority_queue<PII,vector<PII>,greater<PII>>h;
vector<PII>e[N];
int val[N],pre[N],dist[N];
int cnt[N];//以i为终点的最短路径的条数
int pep[N];//pep[i]表示到达i点积累的队员人数
bool vis[N];
int n,m,s,d;

void dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[s]=0;
    pep[s]=val[s];
    cnt[s]=1;
    h.push({0,s});

    while(h.size())
    {
        auto t=h.top();
        h.pop();

        int u=t.y;

        if(vis[u])continue;
        vis[u]=true;

        for(auto ed:e[u])
        {
            int v=ed.x,w=ed.y;
            if(dist[v]>dist[u]+w)
            {
                dist[v]=dist[u]+w;
                cnt[v]=cnt[u];
                pep[v]=pep[u]+val[v];
                pre[v]=u;//记录v是从u转移过来的
                h.push({dist[v],v});
            }
            else if(dist[v]==dist[u]+w)
            {
                cnt[v]+=cnt[u];
                if(pep[v]<pep[u]+val[v])
                {
                    pep[v]=pep[u]+val[v];
                    pre[v]=u;
                }
            }

        }
    }
}

int main()
{
    cin>>n>>m>>s>>d;

    for(int i=0;i<n;i++)
    {
        cin>>val[i];
    }

    for(int i=1;i<=m;i++)
    {
        int a,b,w;
        cin>>a>>b>>w;
        e[a].push_back({b,w});
        e[b].push_back({a,w});
    }

    dijkstra();

    cout<<cnt[d]<<' '<<pep[d]<<endl;
    vector<int>ans;
    for(int i=d;i!=s;i=pre[i])
    {
        ans.push_back(i);
    }

    reverse(ans.begin(),ans.end());

    printf("%d",s);

    for(auto it=ans.begin();it!=ans.end();it++)
    {
        printf(" %d",*it);
    }

    return 0;
}