UVA 3902 <无根树转变成有根树> <根据深度寻找k级祖先>

</pre><pre name="code" class="html">#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <climits>
#include <cstdlib>
#include <ctime>
using namespace std;
//ios::sync_with_stdio(false);
int n,s,k;
vector<int> gr[1005];
vector<int> node[1005];
bool vis[1005];
int fa[1005];

void dfs(int u,int f,int d)
{
	fa[u] = f;			//dfs中可以确定他的双亲
	int nc = gr[u].size();			//与叶子结点相关的只有一个节点那就是他的父节点
	if(nc == 1 && d > k) node[d].push_back(u);    //判断是不是深度大于k的叶子，只有大于k的叶子才能放到node里面去

	for(int i = 0; i < nc;i++) 
	{
		int v = gr[u][i];
		if(v!=f)		//在所有与之相关的结点中除了他的子树还有他的父亲
			dfs(v,u,d+1);
	}
}

void dfs2(int u,int f,int d)
{
	vis[u] = true; //深度小于的话都会被覆盖掉
	int nc = gr[u].size();
	if(d < k)
	for(int i = 0; i < nc;i++)
	{
		int v = gr[u][i];
		if(v != f)
			dfs2(v,u,d+1);
	}
}
int solve()
{
	memset(vis,0,sizeof(vis));
	int ans = 0;
	for(int d = n-1;d > k;d--)  //所有深度大于k的叶子都在node里面，并根据深度存放在node中。
	{
		int nc = node[d].size(); 

		for(int j = 0; j < nc;j++)
		{
			int nq = node[d][j]; 
			if(vis[nq]) continue;
			
			int v = nq;
			for(int q = 0;q < k;q++) v = fa[v]; //fa[i] 是i的双亲
			dfs2(v,-1,0);
			ans++;
			
		}
	}
	return ans;
}
int main()
{
	int t;
	cin >> t;
	while(t--)
	{
		scanf("%d%d%d",&n,&s,&k);
		for(int i = 0;i <= n;i++)
		{
			gr[i].clear();
			node[i].clear();
		}
		for(int i = 0;i < n-1;i++)
		{
			int x,y;
			cin >> x >> y;
			gr[x].push_back(y);
			gr[y].push_back(x);
		}
		dfs(s,-1,0);
		cout << solve() << endl;
	}
	return 0;
}

这题的思路是每次找到一个服务器的话就把在这个服务器能覆盖到的叶子结点全部覆盖掉，然后在那些没有被覆盖到的结点中从找到最大的度数的结点把他的k级祖先作为一个服务器然后循环就可以啦。

点此看题