hdu 4607 Park Visit（树上最长链）

求树上最长链：两遍搜索。

第一次从树上任意点开始，最远点必然是某一条最长链上的端点u。

第二次从u开始，最远点即该最长链的另一端点。

先在最长链上走，不足再去走支链。

把询问数m错打成n，狠狠wa了一次= =

#include<stdio.h>
#include<string.h>

const int MAXN=111111;

struct E{
    int v,next;
}e[MAXN<<1];

struct Q{
    int p,c;
}q[MAXN];

int tol;
int head[MAXN];
int vis[MAXN];

void init()
{
    tol=0;
    memset(head,-1,sizeof(head));
}

void add(int u,int v)
{
    e[tol].v=v;
    e[tol].next=head[u];
    head[u]=tol++;
}

int BFS(int x)
{
    int r,l,i;
    int u,c;
    r=l=0;
    memset(vis,0,sizeof(vis));
    vis[x]=1;
    q[r].p=x;
    q[r++].c=1;
    while(l<r)
    {
        u=q[l].p;
        c=q[l++].c;
        for(i=head[u];i!=-1;i=e[i].next)
        {
            int v=e[i].v;
            if(!vis[v]){
                vis[v]=1;
                q[r].p=v;
                q[r++].c=c+1;
            }
        }
    }
    return u;
}

int main()
{
    int T,n,m,i;
    int u,v,w;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        init();
        for(i=1;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        u=BFS(1);
        v=BFS(u);
        for(i=0;i<m;i++)
        {
            scanf("%d",&w);
            if(q[n-1].c>=w)
                printf("%d
",w-1);
            else
                printf("%d
",(q[n-1].c-1)+(w-q[n-1].c)*2);
        }
    }
    return 0;
}