1087. All Roads Lead to Rome (30)

#include <iostream>
#include <algorithm>
#include <map>
#include <vector>
using namespace std;
int n, k;
const int inf = 99999999;
const int msize = 201;
int e[msize][msize], weight[msize], dis[msize];
bool visited[msize];
vector<int> pre[msize], temppath, path;
map<string, int> ma;
map<int, string> mb;
int maxvalue = 0, mindepth, cntpath = 0;
double maxavg;
//滿足 最小代價->最大幸福->最大平均幸福
void dfs(int v){
    temppath.push_back(v);
    if(v == 1){
        int value = 0;
        for (int i = 0; i < temppath.size(); i++)
            value += weight[temppath[i]];
        double tempavg = 1.0 * value / (temppath.size() - 1);
        if (value > maxvalue) {
            maxvalue = value;
            maxavg = tempavg;
            path = temppath;
        }else if(value == maxvalue && tempavg > maxavg){
            maxavg = tempavg;
            path = temppath;
        }
        temppath.pop_back();
        cntpath++;
        return;
    }
    for (int i = 0; i < pre[v].size(); i++)
        dfs(pre[v][i]);
    temppath.pop_back();
}
int main() {
    fill(e[0], e[0] + msize * msize, inf);
    fill(dis, dis + msize, inf);
    cin >> n >> k;
    string s;
    cin >> s;
    ma[s] = 1;
    mb[1] = s;
    for (int i = 1; i < n; i++) {
        cin >> s >> weight[i+1];
        ma[s] = i + 1;
        mb[i+1] = s;
    }
    string sa, sb;
    int temp;
    for (int i = 0; i < k; i++) {
        cin >> sa >> sb >> temp;
        e[ma[sa]][ma[sb]] = temp;
        e[ma[sb]][ma[sa]] = temp;
    }
    //dijkstra
    dis[1] = 0;
    for (int i = 0; i < n; i++) {
        int u = -1, minn = inf;
        for (int j = 1; j <= n; j++) {
            if(visited[j] == false && dis[j] < minn){
                u = j;
                minn = dis[j];
            }
        }
        if (u == -1) break;
        visited[u] = true;
        for (int v = 1; v <= n; v++) {
            if (visited[v] == false && e[u][v] != inf) {
                if (dis[u] + e[u][v] < dis[v]) {
                    dis[v] = dis[u] + e[u][v];
                    pre[v].clear();
                    pre[v].push_back(u);
                }else if(dis[u] + e[u][v] == dis[v]){
                    pre[v].push_back(u);
                }
            }
        }
    }
    int rom = ma["ROM"];
    dfs(rom);
    printf("%d %d %d %d\n", cntpath, dis[rom], maxvalue, (int)maxavg);
    for (int i = (int)path.size() - 1; i >= 1; i--)
        cout << mb[path[i]] << "->";
    cout << "ROM\n";
    return 0;
}