PTA 数据结构与算法题目集（中文） 7-35 城市间紧急救援（25 分） 迪杰斯特拉算法

7-35 城市间紧急救援（25 分）

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式:

输入第一行给出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 <cstring>
#include <algorithm>
#include <iostream>
#include <vector>
#include <stack>
using namespace std;
const int maxn=505;
const int INF=0x3f3f3f3f;
int n,m,s,d;
int sp[maxn]; //救援人数
int dj[maxn]; //距离起点最短距离
int vis[maxn];//是否列入集合
int road[maxn];//最短路径条数
int path[maxn];//前缀节点
int total[maxn];//总救援人数
struct city
{
    int e;
    int len;
};
vector <city> ve[maxn];
//初始化
void init ()
{
    memset (vis,0,sizeof(vis));
    memset (total,0,sizeof(total));
    memset (road,0,sizeof(road)); 
    memset (path,-1,sizeof(path));
    for (int i=0;i<n;i++)
        dj[i]=INF;
    dj[s]=0;
    total[s]=sp[s];
    road[s]=1;
}
void djst ()
{
    while (1)
    {
       int maxx=INF;
       int u=-1;
       for (int i=0;i<n;i++)
            if(dj[i]<maxx&&!vis[i])
            {
               maxx=dj[i];
               u=i;
            }
       if(u==-1)
          break;
       vis[u]=1;
       for (int i=0;i<ve[u].size();i++)
       {
           int v=ve[u][i].e;
           if(!vis[v])
           {
               if(dj[v]>dj[u]+ve[u][i].len)
               {
                  dj[v]=dj[u]+ve[u][i].len;
                  path[v]=u;
                  road[v]=road[u];
                  total[v]=total[u]+sp[v];

               }
               else if(dj[v]==dj[u]+ve[u][i].len)
               {
                   //这里要写到if语句外面
                    road[v]+=road[u];
                    if(total[u]+sp[v]>total[v])
                    {
                       path[v]=u;
                       total[v]=total[u]+sp[v];
                    }
               }
           }
       }
    }
}
//输出函数
void output()
{
    printf("%d %d\n",road[d],total[d]);
    int temp=d;
    stack<int>ss;
    while (path[temp]!=-1)
    {
        ss.push(temp);
        temp=path[temp];
    }
    ss.push(s);
    while (!ss.empty())
    {
        printf("%d%c",ss.top(),ss.size()==1? '\n':' ');
        ss.pop();
    }
}
int main()
{
    scanf("%d%d%d%d",&n,&m,&s,&d);
    for (int i=0;i<n;i++)
        scanf("%d",&sp[i]);
    init();
    for (int i=0;i<m;i++)
    {
        int x,y,len;
        scanf("%d%d%d",&x,&y,&len);
        city temp1,temp2;
        temp1.e=y; temp1.len=len;
        temp2.e=x; temp2.len=len;
        ve[x].push_back(temp1);
        ve[y].push_back(temp2);
    }
    djst();
    output();
    return 0;
}