【PAT甲级】1087 All Roads Lead to Rome 条条大路通罗马

PAT原题网址PTA | 程序设计类实验辅助教学平台 (pintia.cn)

ACwing原题网址1577. 条条大路通罗马 - AcWing题库

分析：

1. 此题是dijkstra的扩展版本，要用到哈希表来映射城市与编号，以此来找到编号对应城市的幸福指数w[i],

2. 还要输出最小路径，则需要pre[N] 存储 当前结点的前一个结点，把它存进vector<int>path中，最后倒序输出

3. 输出最短路的个数， 最短距离， 最大幸福度（点权值最大）， 平均幸福指数（平均点权值）

AC代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <unordered_map>
#include <vector>

using namespace std;

const int N = 300;

int d[N][N], w[N]; // 存储图 点权值
int dist[N], cnt[N], hy[N], sum[N], pre[N];
//  dist数组 最短路数量 总幸福值 经过的点的最少个数 某个点的前一个点
bool st[N];
int n, m;

unordered_map<string, int>mp;
string city[N];

void dijkstra()
{
    memset(dist, 0x3f, sizeof dist); // dijkstra的第一步 初始化 
    
    dist[1] = 0; // 城市的映射下标是从1 开始的  所以dist[1] = 0 
    cnt[1] = 1; // 必有一条最短路
    
    for(int i = 0; i < n; i ++ )
    {
        int t = -1;
        for(int j = 1; j <= n; j ++ )
            if(st[j] == false && (t == -1 || dist[t] > dist[j]) )   
                t = j;
        
        
        st[t] = true;
        
        for(int j = 1; j <= n; j ++ )
        {
            if(dist[j] > dist[t] + d[t][j])
            {
                dist[j] = dist[t] + d[t][j];
                cnt[j] = cnt[t];
                sum[j] = sum[t] + 1;
                hy[j] = hy[t] + w[j];
                pre[j] = t;
            }
            else if(dist[j] == dist[t] + d[t][j])
            {
                cnt[j] += cnt[t];
                if( hy[j] < hy[t] + w[j])
                {
                    hy[j] = hy[t] + w[j];
                    sum[j] = sum[t] + 1;
                    pre[j] = t;
                }
                else if( hy[j] == hy[t] + w[j])
                {
                    if( sum[j] > sum[t] + 1)
                    {
                        sum[j] = sum[t] + 1;
                        pre[j] = t;
                    }
                }
            }
        }
        
    }
    
}

int main()
{
    memset(d, 0x3f, sizeof d); // 先初始化 
   
    cin >>  n >>  m >> city[1]; // 城市数 道路数 起点城市 
    mp[ city[1] ] = 1; // 对起点城市映射为数字1 
    
    for(int i = 2; i <= n; i ++ )
    {
        cin >> city[i] >> w[i]; 
        mp[city[i]] = i; // 第二个城市映射为数字2 
    }
    
    while(m -- ) // 道路数 
    {
        string x, y;
        int c;
        cin >> x >> y >> c; // 输入城市名字 x y  和 x y间的距离
        int a = mp[x], b = mp[y]; // 找到 x y 对应的数字 
        d[a][b] = d[b][a] = min(d[a][b], c);  
    }
    
    dijkstra();
    int T = mp["ROM"]; // "ROM"  对应的数字为 T
    
    // 输出最短路数量       最短距离          总幸福值         平均幸福值
    cout << cnt[T] << ' ' << dist[T] << ' ' << hy[T] << ' ' << hy[T] / sum[T];
    cout << endl;
    
    vector<int> path;
    for(int i = T; i != 1; i = pre[i]) path.push_back(i); //正序输入  
    
    cout << city[1]; // 起点城市
    for(int i = path.size() - 1; i >= 0; i --)  
        cout << "->" << city[path[i]]; // 倒序输出 找到下标对应的城市名字
        
    cout << endl;
    
    
    return 0;
}