floyd算法简单练习：acwing1471. 牛奶工厂

牛奶生意正红红火火！

农夫约翰的牛奶加工厂内有 NN 个加工站，编号为 1…N，以及 N−1条通道，每条连接某两个加工站。（通道建设很昂贵，所以约翰选择使用了最小数量的通道，使得从每个加工站出发都可以到达所有其他加工站）。

为了创新和提升效率，约翰在每条通道上安装了传送带。

不幸的是，当他意识到传送带是单向的已经太晚了，现在每条通道只能沿着一个方向通行了！

所以现在的情况不再是从每个加工站出发都能够到达其他加工站了。

然而，约翰认为事情可能还不算完全失败，只要至少还存在一个加工站 i 满足从其他每个加工站出发都可以到达加工站 i。

注意从其他任意一个加工站 j 前往加工站 i 可能会经过 i 和 j 之间的一些中间站点。

请帮助约翰求出是否存在这样的加工站 i。

输入格式

输入的第一行包含一个整数 N，为加工站的数量。

以下 N−1行每行包含两个空格分隔的整数 ai 和 bi，满足 1≤ai,bi≤N 以及 ai≠bi。

这表示有一条从加工站 ai 向加工站 bi 移动的传送带，仅允许沿从 ai 到 bi 的方向移动。

输出格式

如果存在加工站 i 满足可以从任意其他加工站出发都可以到达加工站 i，输出最小的满足条件的 i。

否则，输出 −1。

数据范围

1≤N≤100

#include<bits/stdc++.h>
using namespace std;
const int N=110;
//int w[N][N];
int dis[N][N];
bool st[N];
int n,m;
void solve()
{
   // memset(dis,0x3f,sizeof(dis));
    cin>>n;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(i!=j){
                dis[i][j]=0x3f3f3f3f;
            }
        }
    }
    for(int i=1;i<n;i++){
        int a,b;
        cin>>a>>b;
        dis[b][a]=1;
    }
    int p=0;
    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]);
                p++;
            }
        }
    }
    bool flag=0;
    for(int i=1;i<=n;i++){
        int o=0;
        for(int j=1;j<=n;j++){
            if(dis[i][j]==0x3f3f3f3f){
                continue;
            }
            o++;
            if(o==n){
                cout<<i;
                flag=1;
            }
        }
        if(flag){
            break;
        }
    }
    if(!flag){
        cout<<"-1";
    }
    
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    int t=1;
    while(t--){
        solve();
    }return 0;
}