tjut 4607

#include <iostream>  
#include <string.h>  
#include <stdio.h>  
  
using namespace std;  
  
const int N=200010;  
  
int head[N],to[N],next[N],w[N];  
int dis[N],que[N];  
bool vis[N];  
int edge,m,n;  
  
void init()  
{  
    memset(head,-1,sizeof(head));  
    edge=0;  
}  
  
void add(int u,int v,int c)  
{  
    to[edge]=v,w[edge]=c,next[edge]=head[u],head[u]=edge++;  
    to[edge]=u,w[edge]=c,next[edge]=head[v],head[v]=edge++;  
}  
  
void bfs(int s)  
{  
    memset(vis,0,sizeof(vis));  
    memset(dis,0,sizeof(dis));  
    int l,r,v,u;  
    l=r=0;  
    vis[s]=1;  
    dis[s]=0;  
    que[r++]=s;  
    while(r>l)  
    {  
        u=que[l++];  
        for(int i=head[u]; ~i; i=next[i])  
        {  
            if(!vis[v=to[i]])  
            {  
                vis[v]=1;  
                dis[v]=dis[u]+w[i];  
                que[r++]=v;  
            }  
        }  
    }  
}  
  
int treediameter(int s)  
{  
    int u,maxl;  
    bfs(s);  
    maxl=0,u=s;  
    for(int i=1; i<=n; i++)  
        if(dis[i]>maxl)  
            u=i,maxl=dis[i];  
    bfs(u);  
    maxl=0;  
    for(int i=1; i<=n; i++)  
        if(dis[i]>maxl)  
            maxl=dis[i];  
    return maxl;  
}  
  
int main()  
{  
    int u,v,d=1,t,i,j,x;  
    scanf("%d",&t);  
    while(t--)  
    {  
        init();  
        scanf("%d%d",&n,&m);  
        for(i=1;i<=n-1;i++)  
        {  
            scanf("%d%d",&u,&v);  
            add(u,v,1);  
        }  
        int ans=treediameter(1);  
        ans++;  
        while(m--)  
        {  
            scanf("%d",&x);  
            if(x<=ans) printf("%d\n",x-1);  
            else       printf("%d\n",(x-ans)*2+ans-1);  
        }  
    }  
    return 0;  
}  

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值