2020.2.22GDUT寒假训练排位赛2-F

F — Milk Factory

题目大意：

牛奶生意兴隆!农夫约翰牛奶加工厂由N个加工站组成，方便地编号为1…N(1≤N≤100)和N - 1条走道，每条走道连接两个加工站。(人行道很贵，所以农民约翰选择使用最少的人行道，这样人们最终可以从其他车站出发到达任意车站)。为了提高效率，农夫约翰在每个走道上安装了一个传送带。不幸的是，他太晚意识到每个传送带只能单向移动，所以现在沿着每个通道只能单向移动!现在，一个人从一个车站到另一个车站的情况已经不存在了。然而，农民约翰认为，只要有至少一个站i，所有的可能都不会丢失，这样一个人最终可以从其他站到站i。请帮助农民John弄清楚是否存在这样一个站i。
输入
第一行包含一个整数N，即处理站的数量。
接下来的N - 1行每一行都包含两个空格分隔的整数ai和bi, 1≤ai,bi≤N且ai≠bi。这表明有一个传送带从ai站移动到bi站，只允许从ai站移动到bi站。
输出
如果存在一个i站，使一个人可以从任何其他站走到i站，则输出最小的i，否则输出为- 1。

题目分析：

由于数据少，运用Floyd算法经典的3次for循环，再遍历每个点的入度即可。

代码实现：

#include <iostream>
#include <cstdio>
using namespace std;

const int INF = 1111111;
int n;
int dis[105][105] = {0};

void init()
{
    for(int i=0; i<105; i++){
        for(int j=0; j<105; j++){
            dis[i][j] = INF;
            if(i==j) dis[i][j] = 0;
        }
    }
}

int main()
{
    init();
    cin >> n;
    int u,v;
    for(int i=0; i<n-1; i++){
        scanf("%d%d",&u,&v);
        dis[u][v] = 1;
    }
    for(int k=1; k<=n; k++){
        for(int i=1; i<=n; i++){
            for(int j=1; j<=n; j++){
                dis[i][j] = min(dis[i][j],dis[i][k]+dis[k][j]);
            }
        }
    }
    int ans = -1;
    for(int i=1; i<=n; i++){
        int num = 0;
        for(int j=1; j<=n; j++){
            if(dis[j][i]!=INF){
                num++;
            }
        }
        if(num==n){
            ans = i;
            break;
        }
    }
    printf("%d\n",ans);

    return 0;
}