7-8 哈利·波特的考试

https://pintia.cn/problem-sets/15/problems/716

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

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

输入格式:

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

输出格式:

输出哈利·波特应该带去考场的动物的编号、以及最长的变形魔咒的长度，中间以空格分隔。如果只带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

提交代码

#include <stdio.h>
#include <stdlib.h>
#include <iostream>
using namespace std;

#define MAXN 101
#define INF 0x3f3f3f3f

//0x3f3f3f3f = 1061109567
//1061109567 * 2 = 2122219134 < 2147483647
//0x4f4f4f4f = 1330597711
//1330597711 * 2 = 2661195422 > 2147483647 越界 

int arr[MAXN][MAXN] = {0};
int animNum,magicNum;
int maxDist[MAXN] = {0};

void FloydInit(int arr[MAXN][MAXN], int len)
{
    for(int i=1;i<=len;i++)
    {
        for(int j=1;j<=len;j++)
        {
            if(i==j)
                arr[i][j] = 0;
            else
                arr[i][j] = INF;
        }
    }
}

void Floyd(int arr[MAXN][MAXN], int num)
{
    for(int k=1;k<=num;k++)
        for(int i=1;i<=num;i++)
            for(int j=1;j<=num;j++)
                arr[i][j] = arr[i][j] < arr[i][k] + arr[k][j] ? arr[i][j] : arr[i][k] + arr[k][j];
}

int main()
{
    scanf("%d %d",&animNum,&magicNum);
    FloydInit(arr,animNum);
    
    int a,b,len;
    for(int i=0;i<magicNum;i++)
    {
        scanf("%d %d %d",&a,&b,&len);
        arr[a][b] = len;
        arr[b][a] = len;
    }
    Floyd(arr,animNum);
//    for(int i=1;i<=animNum;i++)
//    {
//        for(int j=1;j<=animNum;j++)
//        {
//            printf("%d\t",arr[i][j]);
//        }
//        printf("\n");
//    }
    for(int i=1;i<=animNum;i++)
    {
        maxDist[i] = -INF-1;
        for(int j=1;j<=animNum;j++)
        {
            if(i==j){
                continue;
            }else if(arr[i][j]==INF){
                maxDist[i] = INF;
                break;
            }else{
                maxDist[i] = arr[i][j] > maxDist[i] ? arr[i][j] : maxDist[i];
            }
        }
    }
//    for(int i=1;i<=animNum;i++)
//    {
//        printf("%d ",maxDist[i]);
//    }
//    printf("\n");
    int minVal = maxDist[1];
    int minIndex = 1;
    for(int i=1;i<=animNum;i++)
    {
        if(maxDist[i] < minVal)
        {
            minVal = maxDist[i];
            minIndex = i;
        }
    }
    if(minVal==INF)
        printf("0\n");
    else
        printf("%d %d\n",minIndex,minVal);
    
    return 0;
}