UVAlive 3092 无根树->有根树 dfs

题目链接：UVALA 3092

题意：给定一个无根树，选其中一个非叶子节点作为服务器， 求要放置最小的服务器数满足 每个叶子节点到最近的服务器的距离最大为k，且服务器只能放在非叶子节点上。

题解：先用dfs把无根树转成以给定服务器为根节点的有根树，那么 深度d <= k的叶子节点就可以不用考虑了，然后 对于 深度大于k的叶子节点  最优解就是 把 服务器设在 他k级祖先上（k倍父节点），然后选择完服务器用dfs覆盖距离这个节点距离k之内的可达叶子节点，直至所有叶子节点均被访问。

AC code：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;

#define debug 0
#define M(a, b) memset(a, b, sizeof(a))

const int maxn = 1000 + 5;

vector<int> gr[maxn], node[maxn];
int n, s, k;
int fa[maxn];
bool vis[maxn];

void dfs(int u, int f, int d) {										//建树
	fa[u] = f;
	int nc = gr[u].size();
	if (nc == 1 && d > k)											//深度大于k叶子节点，记录下来 
		node[d].push_back(u);
	for (int i = 0; i < nc; i++) {
		int v = gr[u][i];
		if (v != f) {												//因为子结点的vector容器内也有父节点，所以这个判断不能省
			dfs(v, u, d + 1);
		}
	}
}
void dfs2(int u, int f, int d) {
	vis[u] = 1;														//标为已访问表示 改点已被覆盖
	int nc = gr[u].size();											//找到服务器 递归覆盖
	/*if (n == 1 && d > k)
		node[d].push_back(u);*/
	for (int i = 0; i < nc; i++) {
		int v = gr[u][i];
		if (v != f && d < k) {										//覆盖范围最大为k
			dfs2(v, u, d + 1);										//递归
		}
	}
}
int solve() {
	int ans = 0;
	for (int d = n - 1; d > k; d--) {								//枚举深度大于k的叶子节点
		for (int i = 0; i < node[d].size(); i++)		
		{			
			int v = node[d][i];
			if (!vis[v]) {											//未被覆盖
				for (int i = 1; i <= k; i++) {					
					v = fa[v];										//找k级祖先
				}
				dfs2(v, -1, 0);					
				ans++;				
			}
		}
	}
		return ans;
}
int main() {
#if debug
	freopen("in.txt", "r", stdin);
#endif //debug

	int t, a, b;
	scanf("%d", &t);
	while (t--) {

		M(fa, 0);
		M(vis, 0);	
		scanf("%d%d%d", &n, &s, &k);
		for (int i = 1; i <= n; i++) {
			gr[i].clear();
			node[i].clear();												//初始化
		}

		for (int i = 1; i < n; i++) {
			scanf("%d%d", &a, &b);
			gr[a].push_back(b);
			gr[b].push_back(a);
		}

		dfs(s, -1, 0);														//递归建树
		//printf("1");

		printf("%d\n", solve());
	}


	return 0;
}