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

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

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

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

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

 #include<stdio.h>
#include<string.h> const int MAXN=; struct E{
int v,next;
}e[MAXN<<]; struct Q{
int p,c;
}q[MAXN]; int tol;
int head[MAXN];
int vis[MAXN]; void init()
{
tol=;
memset(head,-,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=;
memset(vis,,sizeof(vis));
vis[x]=;
q[r].p=x;
q[r++].c=;
while(l<r)
{
u=q[l].p;
c=q[l++].c;
for(i=head[u];i!=-;i=e[i].next)
{
int v=e[i].v;
if(!vis[v]){
vis[v]=;
q[r].p=v;
q[r++].c=c+;
}
}
}
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=;i<n;i++)
{
scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
u=BFS();
v=BFS(u);
for(i=;i<m;i++)
{
scanf("%d",&w);
if(q[n-].c>=w)
printf("%d\n",w-);
else
printf("%d\n",(q[n-].c-)+(w-q[n-].c)*);
}
}
return ;
}
04-15 14:33