n台机器组成一个树状网络,叶结点为客户端,其它结点为服务器。目前有一台服务器提供VOD服务。要求尽可能少的在其它服务器上装VOD服务使得每个客户端到最近的提供VOD服务的服务器的距离不超过一个整数k。
输入格式:
输入第一行为数据组数T。每组数据第一行为树中的结点数n(3~1000);下一行包含两个整数s和k(1~n,1~n),其中s为已经提供VOD服务的服务器的编号,k是叶子和提供VOD服务的服务器的距离上限;以下n-1行每行包含两个整数,即树的一边。
输出格式:
对每组数据,输出一个整数,即还需要几台提供VOD服务的服务器。
思路:将已经提供VOD服务的服务器当作根结点+DFS
代码如下:
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int maxn = 1010;
vector<int> edge[maxn],nodes[maxn];
int n,s,k,fa[maxn],visit[maxn];
void DFS1(int now,int f,int d)
{
fa[now] = f;
int nc = edge[now].size();
if (nc == 1 && d > k)//找到不能享受VOD服务的叶子
nodes[d].push_back(now);
for (int i = 0; i < nc; i++)
{
int next = edge[now][i];
if (next != f)
DFS1(next,now,d + 1);
}
}
void DFS2(int now,int f,int d)
{
visit[now] = 1;
int nc = edge[now].size();
for (int i = 0; i < nc; i++)
{
int next = edge[now][i];
if (next != f && d < k)//只访问距离小于k的叶子
DFS2(next,now,d + 1);
}
}
int result()
{
int count = 0;
memset(visit,0,sizeof(visit));
for (int d = n - 1; d > k; d--)
for (int i = 0; i < nodes[d].size(); i++)
{
int now = nodes[d][i];
if (visit[now])//如果已被访问
continue;
int v = now;
for (int j = 0; j < k; j++)
v = fa[v];
DFS2(v,-1,0);//在结点v处提供VOD服务
count++;
}
return count;
}
int main()
{
int T;
scanf("%d",&T);
while (T--)
{
scanf("%d%d%d",&n,&s,&k);
for (int i = 1; i <= n; i++)//清空
{
edge[i].clear();
nodes[i].clear();
}
for (int j = 0; j < n - 1; j++)
{
int a,b;
scanf("%d%d",&a,&b);//输入边
edge[a].push_back(b);
edge[b].push_back(a);
}
DFS1(s,-1,0);//保存不能享受VOD服务的叶子
printf("%d\n",result());
}
return 0;
}