#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;
const int sz=1e5+3;
int vis[sz],dis[sz],head[sz];
struct node
{
int v,next;
}g[2*sz];
int n,m,ans,node,cnt;
void add(int u,int v)
{
cnt++;
g[cnt].v=v;
g[cnt].next=head[u];
head[u]=cnt;
}
void init()
{
memset(vis,0,sizeof(vis));
memset(dis,0,sizeof(dis));
}
void dfs(int u,int t)
{
vis[u]=1;
int i;
for(i=head[u];i!=-1;i=g[i].next)
{
int v=g[i].v;
if(!vis[v])
{
dis[v]=t+1;
if(dis[v]>ans)
{
ans=dis[v];
node=v;
}
dfs(v,dis[v]);
}
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
init();
memset(head,-1,sizeof(head));
scanf("%d %d",&n,&m);
int i,j;
ans=0;
cnt=0;
for(i=1;i<n;i++)
{
int a,b;
scanf("%d %d",&a,&b);
add(a,b);
add(b,a);
}
dfs(1,0);
ans=0;
init();
dfs(node,0);
while(m--)
{
int k;
scanf("%d",&k);
if(k<=ans+1)
printf("%d\n",k-1);
else
printf("%d\n",ans+(k-ans-1)*2);
}
}
return 0;
}
求树的直径
最新推荐文章于 2021-07-26 09:37:57 发布