题意:n台及其连成一个树状网络,其中叶结点是客户端,其他结点是服务器。已经有一台服务器提供VOD服务,要求出最少在几个记得点放置服务器,使得所有客户端到服务器的距离都不超过k。
思路:先将无根树状化为有根树。之后就利用贪心,找到深度最大的结点u,每次放置服务器的地方都要放在u的k级祖先,这样是最划算的。
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <vector> using namespace std; const int MAXN = 1005; vector<int> gr[MAXN], nodes[MAXN]; int n, s, k, 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) nodes[d].push_back(u); //利用nodes表可以省去对于叶子结点深度的搜索 for (int i = 0; i < nc; i++) { int v = gr[u][i]; if (v != f) dfs(v, u, d + 1); } } void dfs_cover(int u, int f, int d) { vis[u] = true; int nc = gr[u].size(); for (int i = 0; i < nc; i++) { int v = gr[u][i]; if (v != f && d < k) dfs_cover(v, u, d + 1); } } int solve() { int ans = 0; memset(vis, 0, sizeof(vis)); for (int d = n - 1; d > k; d--) for (int i = 0; i < nodes[d].size(); i++) { int u = nodes[d][i]; if (vis[u]) continue; int v = u; for (int j = 0; j < k; j++) v = fa[v]; dfs_cover(v, -1, 0); ans++; } return ans; } int main() { int cas; scanf("%d", &cas); while (cas--) { scanf("%d%d%d", &n, &s, &k); for (int i = 1; i <= n; i++) { gr[i].clear(); nodes[i].clear(); } for (int i = 0; i < n - 1; i++) { int a, b; scanf("%d%d", &a, &b); gr[a].push_back(b); gr[b].push_back(a); } dfs(s, -1, 0); int ans = solve(); printf("%d\n", ans); } return 0; }