例题1.15 UVALive-3902 树上的搜索

传送门

题目大意：n台机器连成一个树状网络，叶节点是客户端，其他节点是服务器，一开始只有一台服务器提供一种服务，但它只能覆盖k的距离内的客户端，你让尽量少的服务器提供这项服务，使得所有的客户端都被覆盖。

解题思路：我们将这个无根树转换为有根树，那么提供服务的服务器就是根节点，那么k深度以内的叶节点都被覆盖了，因此我们只要找k以下的即可。从最深的节点开始找，它需要被覆盖，距离它k距离的服务器提供服务是最经济的，因此我们只要找的它的k级祖先即可，那么这个服务器周围的k的距离都被覆盖，都遍历一次即可。

AC代码：

#include <iostream>
#include <cstdio>
#include <vector>
#include <cstring>

using namespace std;
const int maxn = 1000+5;

int n, s, k;
vector<int> List[maxn], Nodes[maxn]; //List链接图， Nodes保存每一层的节点

int fa[maxn];// 记录父节点
bool covered[maxn];

void dfs_deep(int c, int f, int d) //将无根树转换为有根树，并找出所有的叶子节点，记录深度
{
	fa[c] = f;
	int cn = List[c].size();
	if(cn == 1 && d>k) Nodes[d].push_back(c);//叶子
	for(int i=0; i<cn; i++)
		if(List[c][i] != f) dfs_deep(List[c][i], c, d+1);
}

void dfs(int c, int f, int d)//覆盖所有能覆盖的节点
{
	covered[c] = 1;
	int cn = List[c].size();
	for(int i=0; i<cn; i++)
	{
		int v = List[c][i];
		if(d<k && v!= f) dfs(v, c, d+1);
	}
}

int solve()
{
	int ans = 0;
	memset(covered, 0, sizeof(covered));
	for(int d=n-1; d>k; d--)//从最深的节点开始找，找到k层以下，因为k层之上的都被根覆盖了
	{
		int cn = Nodes[d].size();
		for(int j=0; j<cn; j++)
		{
			int cur = Nodes[d][j];
			if(!covered[cur]) //次节点没有被覆盖
			{
				int f = cur;
				for(int i = 0; i<k; i++) f = fa[f];
				dfs(f, -1, 0);
				ans++;
			}
		}
	}
	return ans;
}

int main()
{
	int t;
	scanf("%d", &t);
	while(t--)
	{
		for(int i=0; i<maxn; i++)
		{
			List[i].clear();
			Nodes[i].clear();
		}
		scanf("%d%d%d", &n, &s, &k);
		int a, b;
		for(int i=0; i<n-1; i++)
		{
			scanf("%d%d", &a, &b);
			List[a].push_back(b);
			List[b].push_back(a);
		}
		dfs_deep(s, -1, 0);
		printf("%d\n", solve());
	}
	return 0;
}