TopCoder SRM 670 Div2 Problem 1050 - Treestrat (floyd)

题意

在一棵树上，一些猫在一些结点上，一些老鼠在一些结点上。
每轮老鼠方先走，每只老鼠可以选择原地不动或者走到相邻结点。
如果老鼠走到有猫上的结点游戏结束。
问最多进行几轮游戏

思路

这么考虑。
每只老鼠的使命是尽可能地活得久。所以我们可以枚举他的终点，也就是树上的全部结点。
因为全部的猫都有可能来围剿这只老鼠，所以我们要看看有没有猫能在他之前到达这个结点，如果能的话，他就去不了。

对于全部的老鼠都进行枚举，维护一下它们中能去的距离的最小值，意思就是猫就去捉这只老鼠了。这样我们就能得到答案。

代码

class Treestrat {
public:
    int n, mp[100][100];

    void floyd()
    {
        for (int k = 0; k < n; k++)
            for (int i = 0; i < n; i++)
                for (int j = 0; j < n; j++) mp[i][j] = min(mp[i][j], mp[i][k] + mp[k][j]);
    }

    int roundcnt(vector<int> tree, vector<int> A, vector<int> B) {
        MS(mp, INF);
        for (int i = 0; i < SZ(tree); i++) mp[tree[i]][i+1] = mp[i+1][tree[i]] = 1;
        for (int i = 0; i < SZ(tree)+1; i++) mp[i][i] = 0;
        n = SZ(tree)+1;
        floyd();
        int ret = INF;
        for (int u : A)
        {
            int ans = 0;    //当前老鼠最多能跑多久
            for (int tar = 0; tar < n; tar++)
            {
                int cur_run = INF;
                bool flag = true;
                for (int v : B)
                {
                    if (mp[u][tar] >= mp[v][tar]) flag = false;
                    else cur_run = min(cur_run, mp[v][tar]);
                }
                if (flag) ans = max(ans, cur_run);
            }
            ret = min(ret, ans);
        }
        return ret;
    }
};