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;
}



你可能感兴趣的:(uva 1267 - Network(dfs))