树的直径
求出树的直径后分类讨论。
#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;
}