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