uva 1267 - Network(dfs)

题目大意：给出n，s，k，即有n台客户端和服务器（边界为客户端），然后给出n-1条边，连接n台终端，现在在序号为s的服务器上有一个资源，为了方便所有用户浏览（访问路径长度不超过k），要在较少的服务器上复制资源，问说最少再在几个服务器上复制资源。

解题思路：首先现将无根图转化成树，以资源所在的服务器s为根，然后dfs建树，建树的过程中将所有叶子节点按照深度记录。其二就是枚举需要复制资源的服务器，从节点深度最大的叶子节点开始，向上查找父亲节点k次，即为当前的最有策略，然后将距离该服务器不超过k的终端全部标记。

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

vector<int> A[1010], nodes[1010];
int n, s, k, fa[1010], vis[1010];

void DFS(int u, int f, int d) {
fa[u] = f;
int len = A[u].size();
if (len == 1 && d > k)
nodes[d].push_back(u);
for (int i = 0; i < len; i++)
if (A[u][i] != f)
DFS(A[u][i], u, d+1);
}

void DFS_2(int u, int f, int d) {
vis[u] = 1;
int len = A[u].size();
for (int i = 0; i < len; i++)
if (A[u][i] != f && d < k)
DFS_2(A[u][i], 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;
for (int j = 0; j < k; j++)
u = fa[u];

DFS_2(u, -1, 0);
ans++;
}
return ans;
}

int main() {
int T;
scanf("%d", &T);
while (T--) {
scanf("%d%d%d", &n, &s, &k);
for (int i = 0; i <= n; i++) {
A[i].clear();
nodes[i].clear();
}

for (int i = 0; i < n-1; i++) {
int x, y;
scanf("%d%d", &x, &y);
A[x].push_back(y);
A[y].push_back(x);
}
DFS(s, -1, 0);

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