0
点赞
收藏
分享

微信扫一扫

poj 4607 Park Visit --2013多校联合赛第一场---1008


解题报告:首先如果 k 小于等于直径长度,那么答案为k−1。 如果 k 大于直径长度,设直径长度为 r,那么答案为r−1+(k−r)∗2。 


先找树的最长路;

找树中任意一点,dfs找该点所能达到的最远的点vv,然后从vv点dfs找树的最长路。。


#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;

int t,n,m;
vector<int> q[100010];
int dp[100010],deep[100010];
int vv;
int temp;
int dfs(int u,int fa)
{
	int i,v;
	if(q[u].size()==1&&u!=vv) return dp[u]=1;
	for(i=0;i<q[u].size();i++){
		v=q[u][i];
		if(v!=fa){
			int l=dfs(v,u);
			if(l+1>dp[u])
				dp[u]=l+1;
		}
	}
	return dp[u];
}
void dfs1(int u)
{
	for(int i=0;i<q[u].size();i++){
		int v=q[u][i];
		if(!deep[v]){
		deep[v]=deep[u]+1;
		dfs1(v);
		}
	}
}
int main()
{
	int i,j,k;
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&m);
		for(i=1;i<=n;i++)
			q[i].clear();
		for(i=1;i<n;i++){
			int a,b;
			scanf("%d%d",&a,&b);
			q[a].push_back(b);
			q[b].push_back(a);
		}
		int mx=0;
		memset(dp,0,sizeof(dp));
		memset(deep,0,sizeof(deep));
		deep[1]=1;
		dfs1(1);
		for(i=1;i<=n;i++)
			if(deep[i]>mx){
				mx=deep[i];
				vv=i;
			}
		int r=dfs(vv,-1)-1;
		while(m--){
			int a;
			scanf("%d",&a);
			if(a<=r+1)
				printf("%d\n",a-1);
			else
				printf("%d\n",(a-r-1)*2+r);
		}
	}
}





举报

相关推荐

0 条评论