L2-001. 紧急救援 (Dijkstra最短路径变形)

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

dij最短路变形 需要记录最短路径条数以及“最大权路径和”及打印其路径 双限制条件(最短路径 最大权路径和)

#include <bits/stdc++.h>

using namespace std;

const int inf = 0x3f3f3f3f;

bool v[555]; ///标记点收录否
int dis[555]; ///单源最短路长度
int ps[555]; /// 路径path
int num[555],sum[555];  ///最短路条数 最多的救援人数
int mp[555][555]; ///记录距离
int peo[555]; ///记录人数
int n; ///节点0~n

void dij(int s,int ed)
{
    int i,j,d,pos;
    memset(v,0,sizeof(v));
    memset(ps,-1,sizeof(ps));
    memset(dis,inf,sizeof(dis));

    dis[s] = 0;
    for(i=1; i<n; i++)
    {
        d = inf;
        for(j=0; j<n; j++)
        {
            if(!v[j] && dis[j] < d)
            {
                d = dis[j];
                pos = j;
            }
        }
        v[pos] = 1;
        for(j=0; j<n; j++)
        {
            if(!v[j] && dis[j] > dis[pos] + mp[pos][j])
            {
                dis[j] = dis[pos] + mp[pos][j];
                ps[j] = pos;
                num[j] = num[pos];  ///显然 j与pos的最短路径条数一样多 因为是一条路
                sum[j] = sum[pos] + peo[j];  ///增加人数 pos->j
            }
            else if(!v[j] && dis[j] == dis[pos] + mp[pos][j])
            {
                if(sum[j] < sum[pos] + peo[j])  ///更新最多的人数
                {
                    sum[j] = sum[pos] + peo[j];
                    ps[j] = pos;
                }
                num[j] = num[pos] + num[j];  ///这里j的最短路径条数就和pos的不一样了
                ///假设pos有x条 j有y条 那么j的最短路就有x+y条
            }
        }
    }
}

void path(int d) ///打印路径
{
    int i;
    vector<int>ans;
    while(d != -1)
    {
        ans.push_back(d);
        d = ps[d];
    }
    for(i=ans.size()-1; i>0; i--) ///逆序打印
    {
        cout<<ans[i]<<" ";
    }
    cout<<ans[i]<<endl;
}

int main()
{
    int i,j,s,d;
    int m,x,z,y;
    std::ios::sync_with_stdio(0);
    while(cin>>n>>m>>s>>d)
    {
        memset(mp,inf,sizeof(mp));
        memset(num,0,sizeof(num));
        memset(sum,0,sizeof(sum));
        for(i=0; i<n; i++) cin>>peo[i];
        for(i=1; i<=m; i++)
        {
            cin>>x>>y>>z;
            if(mp[x][y] > z)
            {
                mp[x][y] = mp[y][x] = z;
            }
        }
        num[s] = 1;     ///
        sum[s] = peo[s]; ///起点初始化
        dij(s,d);
        cout<<num[d]<<" "<<sum[d]<<endl;  ///打印终点记录的数据
        path(d);
    }
    return 0;
}