题目链接: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;
}