数据结构与算法题目集|7-8 哈利·波特的考试 c++满分题解

哈利·波特要考试了，他需要你的帮助。这门课学的是用魔咒将一种动物变成另一种动物的本事。例如将猫变成老鼠的魔咒是haha，将老鼠变成鱼的魔咒是hehe等等。反方向变化的魔咒就是简单地将原来的魔咒倒过来念，例如ahah可以将老鼠变成猫。另外，如果想把猫变成鱼，可以通过念一个直接魔咒lalala，也可以将猫变老鼠、老鼠变鱼的魔咒连起来念：hahahehe。

现在哈利·波特的手里有一本教材，里面列出了所有的变形魔咒和能变的动物。老师允许他自己带一只动物去考场，要考察他把这只动物变成任意一只指定动物的本事。于是他来问你：带什么动物去可以让最难变的那种动物（即该动物变为哈利·波特自己带去的动物所需要的魔咒最长）需要的魔咒最短？例如：如果只有猫、鼠、鱼，则显然哈利·波特应该带鼠去，因为鼠变成另外两种动物都只需要念4个字符；而如果带猫去，则至少需要念6个字符才能把猫变成鱼；同理，带鱼去也不是最好的选择。

输入格式:

输入说明：输入第1行给出两个正整数N (≤100)和M，其中N是考试涉及的动物总数，M是用于直接变形的魔咒条数。为简单起见，我们将动物按1~N编号。随后M行，每行给出了3个正整数，分别是两种动物的编号、以及它们之间变形需要的魔咒的长度(≤100)，数字之间用空格分隔。

输出格式:

输出哈利·波特应该带去考场的动物的编号、以及最长的变形魔咒的长度，中间以空格分隔。如果只带1只动物是不可能完成所有变形要求的，则输出0。如果有若干只动物都可以备选，则输出编号最小的那只。

输入样例:

6 11
3 4 70
1 2 1
5 4 50
2 6 50
5 6 60
1 3 70
4 6 60
3 6 80
5 1 100
2 4 60
5 2 80

输出样例:

4 70

dijkstra算法的应用，本质上是计算最短路径问题，第一个测试点一遍过，后面的无解和输出最小编号的动物分别采用了设flag检测有没有无穷大的距离以及相同最短距离判断是否为编号最小来实现，总的来讲只要对dijkstra算法熟悉这题难度不大，并没有变种

#include <iostream>
#include <vector>
#include <queue>
#include <unordered_map>
#include <climits>
using namespace std;
typedef pair<int, int> pii; // first表示距离，second表示节点索引

// Dijkstra算法
// 接受一个图，一个起始节点和一个存储节点的映射
void dijkstra(unordered_map<int, unordered_map<int, int>>& graph, int start, unordered_map<int, int>& distances) {
    priority_queue<pii, vector<pii>, greater<pii>> pq; // 构建Dijkstra算法的优先队列
    pq.push({0, start});
    distances[start] = 0;
    
    while (!pq.empty()) {
        int dist = pq.top().first;
        int index = pq.top().second;
        pq.pop();
        
        if (dist > distances[index]) continue; // 如果当前节点距离大于已知距离，则跳过节点
        
        for (auto it = graph[index].begin(); it != graph[index].end(); it++) // 遍历所有邻居
        {
            int next_index = it->first;
            int weight = it->second;
            int next_dist = dist + weight; // 计算当前节点到这一邻居节点的新距离
            
            if (next_dist < distances[next_index]) {
                distances[next_index] = next_dist;
                pq.push({next_dist, next_index});
            }
        }
    }
}

int main() {
    int n, m;
    bool flag = true;
    cin >> n >> m;
    unordered_map<int, unordered_map<int, int>> graph;
    
    // 读入图的边
    for (int i = 0; i < m; i++) {
        int animal1, animal2, length;
        cin >> animal1 >> animal2 >> length;
        graph[animal1][animal2] = length;
        graph[animal2][animal1] = length;
    }
    
    unordered_map<int, int> longest_paths;
    for (int animal = 1; animal <= n; animal++) // 从最小编号的动物开始一次执行Dijkstra算法，并算出最短路径的长度
    {
        unordered_map<int, int> distances;
        for (int i = 1; i <= n; i++) distances[i] = INT_MAX; // 初始化所有距离为无限大
        
        dijkstra(graph, animal, distances);
        
        // 执行检查，如果存在到不了的动物直接结束遍历
        for (auto& st : distances) {
            if (st.second == INT_MAX) {
                flag = false;
                break;
            }
        }
        
        if (!flag)
            break;
        
        int longest_path = 0;
        unordered_map<int, int>::iterator it;
        for (it = distances.begin(); it != distances.end(); it++) // 找出最长的一条路径
        {
            if (it->second > longest_path && it->second != INT_MAX) {
                longest_path = it->second;
            }
        }
        longest_paths[animal] = longest_path;
    }
    
    int min_longest_path = INT_MAX;
    int best_animal = -1; // 初始化需要输出的数据
    
    for (auto& path : longest_paths) {
        if (path.second < min_longest_path || (path.second == min_longest_path && path.first < best_animal)) {
            min_longest_path = path.second;
            best_animal = path.first;
        }
    }
    
    if (!flag) {
        cout << "0" << endl;
    } else {
        cout << best_animal << " " << min_longest_path;
    }
    return 0;
}