题意:n台机器连成一个树状网络,其中叶节点是客户端,其他结点是服务器。目前有一台服务器正在提供VOD服务。现在要求在一些其他服务器上也安装VOD服务,使得每台客户端到最近服务器的距离不超过k。求最少还需安装VOD服务的服务器台数。
将无根树转化为有树,题目给出了一个已安装服务的服务器,那么就选它作为根来构造树,这样深度不超过k的结点都不用考虑(已经被根结点所代表的服务器覆盖到了)。我们首先满足深度最大的叶节点,我们选取它的 k 级父节点作为要新增服务的服务器,这样一定是最优的。然后增加一个服务器后,就对这台服务器进行DFS,覆盖与这台服务器距离不超过 k 的所有叶节点,然后再找下一个深度最大且未被覆盖的叶节点进行增加服务器,直到所有叶节点均被覆盖为止。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <map>
#include <set>
using namespace std;
const int MAX = 1005;
int n, s, k;
vector <int> node[MAX], deep[MAX]; //相连节点数组和深度结点数组
int father[MAX]; //父节点数组
bool flag[MAX]; //标记是否已覆盖
void DFS_Father(int x, int pre, int d) //计算父节点
{
father[x] = pre;
if(node[x].size() == 1) //叶节点
deep[d].push_back(x);
for(int i = 0; i < node[x].size(); i++)
{
if(node[x][i] != pre)
DFS_Father(node[x][i], x, d + 1);
}
}
void DFS_Cover(int x, int pre, int d) //覆盖与当前所选服务器距离不超过k的叶节点
{
flag[x] = true;
if(d == k)
return;
for(int i = 0; i < node[x].size(); i++)
{
if(node[x][i] != pre)
DFS_Cover(node[x][i], x, d + 1);
}
}
void input()
{
scanf("%d%d%d", &n, &s, &k);
for(int i = 0; i < MAX; i++)
node[i].clear();
int x, y;
for(int i = 1; i < n; i++)
{
scanf("%d%d", &x, &y);
node[x].push_back(y);
node[y].push_back(x);
}
}
void solve()
{
int ans = 0;
for(int i = 0; i < MAX; i++)
deep[i].clear();
DFS_Father(s, 0, 0);
memset(flag, false, sizeof(flag));
for(int d = n - 1; d > k; d--) //只需考虑深度大于k的结点
{
for(int i = 0; i < deep[d].size(); i++)
{
int temp = deep[d][i];
if(flag[temp]) //已覆盖的结点不用考虑
continue;
for(int j = 0; j < k; j++) //寻找k级祖先
temp = father[temp];
DFS_Cover(temp, 0, 0);
ans++;
}
}
printf("%d\n", ans);
}
int main()
{
int T;
scanf("%d", &T);
while(T--)
{
input();
solve();
}
return 0;
}