PTA 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

//#if 0
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <math.h>
#include <string>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#include <time.h>
#include <set>
#include <list>
#include <iostream>
using namespace std;
#define ll long long
#define ull unsigned long long
#define cls(x) memset(x,0,sizeof(x))
#define clslow(x) memset(x,-1,sizeof(x))
#define INF  0x3f3f3f3f
const int maxn=2000;

int tot=0,num=0,head[maxn],dis[maxn],nod[maxn],team[maxn],dt[maxn],last[maxn];
int n,m,s,d;
/*
dis就是迪杰斯特拉存最短路长度的数组
nod标记已经走过的城市，head是前向星的头数组，
team用来存每个城市救援队人数
dt类似于dis：表示到第i个点的时候最多可以召集多少救援人数
last表示找到的最优解中第i个点上一个是哪一个点
num是最短路条数
tot用于前向星
*/

struct nn
{
    int v,next,w;//v是每条边的结尾节点，next表示下一个u开头的边的存储位置，w表示权值
}l[250000];//500个点，双向边，开500*500个存储空间，否则最后一个样例就是段错误

struct ds
{
    int w,v;//w表示每个边的长度，v表示每条边的结尾节点
    ds(){}
    ds(int w,int v):w(w),v(v){}
    bool operator < (const ds & a) const
    {
        return w>a.w;//优先队列中按边权值从小到大排序
    }
};

priority_queue<ds>q;

void addline(int u,int v,int w)//前向星加边
{
    l[tot].next=head[u];
    l[tot].v=v;
    l[tot].w=w;
    head[u]=tot;
    tot++;
}

void ans(int now,int p)//递归求最短路条数
{
//    cout<<now<<" "<<p<<endl;
    if(p==dis[d] && now==d)num++;
    else{
        for(int i=head[now];i!=-1;i=l[i].next){
//            cout<<" for "<<l[i].v<<" "<<p+l[i].w<<endl;
            if(p+l[i].w<=dis[d])
                ans(l[i].v,p+l[i].w);
        }
    }
}

void dij(int st,int ed)
{
    memset(dis,INF,sizeof(dis));
    memset(nod,0,sizeof(nod));
    memset(last,-1,sizeof(last));
    q.push(ds(0,st));
    dis[st]=0;
    dt[st]=team[st];
    while(!q.empty()){
        ds now=q.top();
        q.pop();
        int u=now.v;
        if(nod[u])continue;
        nod[u]=1;
        for(int i=head[u];i!=-1;i=l[i].next){
            int v=l[i].v;

            if(dis[v]>=dis[u]+l[i].w){//跟纯粹的dij不同地方，当路长一样时，特判救援人数
            //来决定是否换路
                if(dis[v]==dis[u]+l[i].w){
                    if(dt[u]+team[v]>dt[v]){
                        last[v]=u;
                    }
                    dt[v]=max(dt[u]+team[v],dt[v]);
                }
                else{
                    dis[v]=dis[u]+l[i].w;
                    dt[v]=dt[u]+team[v];
                    last[v]=u;
                }
            }
            if(nod[v]==0){
                q.push(ds(l[i].w,v));
            }
        }
    }
//    cout<<1<<endl;
    ans(s,0);
    stack<int>ss;
    for(int i=ed;i!=-1;i=last[i]){
        ss.push(i);
    }
    cout<<num<<" "<<dt[ed]<<endl;
    while(!ss.empty()){
        int i=ss.top();
        ss.pop();
        cout<<i;
        if(!ss.empty())cout<<" ";
    }
}

int main()
{
//    freopen("in1.txt","r",stdin);
//    freopen("out.txt","w",stdout);
    memset(head,-1,sizeof(head));
//由于最后一个样例数很大，不用c输入就超时
    cin>>n>>m>>s>>d;
    for(int i=0;i<n;i++)scanf("%d",&team[i]);
    for(int i=0;i<m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        addline(u,v,w);
        addline(v,u,w);
    }
    dij(s,d);
    return 0;
}