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;    }};