hdu 4607——Park Visit（暑假第一次多校）

树的直径

求出树的直径后分类讨论。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define maxn 200005
int head[maxn];
int v[maxn],next[maxn],cnt;
int n,m,r;
void add(int x,int y)
{
	v[cnt]=y;
	next[cnt]=head[x];
	head[x]=cnt++;
	
	v[cnt]=x;
	next[cnt]=head[y];
	head[y]=cnt++;
}
int bfs(int x)
{
	int dis[maxn];
	int end;
	memset(dis,-1,sizeof(dis));
	queue<int> q;
	q.push(x);
	dis[x]=1;
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		end=u;
		r=dis[u];
		for(int i=head[u];i!=-1;i=next[i])
			if(dis[v[i]]==-1)
			{
				dis[v[i]]=dis[u]+1;
				q.push(v[i]);
			}
	}
	return end;
}
int main()
{
	int t;
	int x,y;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&m);
		cnt=0;
		memset(head,-1,sizeof(head));
		for(int i=0;i<n-1;i++)
		{
			scanf("%d%d",&x,&y);
			add(x,y);
		}
		bfs(bfs(1));
		int tmp;
		while(m--)
		{
			scanf("%d",&tmp);
			if(tmp<=r)
				printf("%d\n",tmp-1);
			else
				printf("%d\n",r-1+(tmp-r)*2);
		}
	}
	return 0;
}