hdu 4607 树的直径

时间:2023-03-10 00:07:05
hdu 4607 树的直径

思路:利用dfs遍历整棵树,找出最长子树与次长子树,两者的和最大就是直径。

若k值小于直径就输出k-1,否则输出(k-d-1)*2+d;

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define Maxn 1000010
using namespace std;
int vi[Maxn],head[Maxn],ans,e;
struct Edge{
int u,v,next,val;
}edge[Maxn];
void init()
{
e=;
memset(head,-,sizeof(head));
ans=;
memset(vi,,sizeof(vi));
}
void add(int u,int v,int val)
{
edge[e].u=u,edge[e].v=v,edge[e].val=val,edge[e].next=head[u],head[u]=e++;
edge[e].v=u,edge[e].u=v,edge[e].val=val,edge[e].next=head[v],head[v]=e++;
}
int dfs(int u)
{
vi[u]=;
int i,j,temp=,Max=,lMax=;
for(i=head[u];i!=-;i=edge[i].next)
{
if(!vi[edge[i].v])
{
temp=dfs(edge[i].v);
if(temp+>=Max)
{
lMax=Max;
Max=temp+;
}
else
{
if(temp+>lMax)
lMax=temp+;
}
if(Max+lMax>ans)
ans=Max+lMax;
}
}
return Max;
}
int main()
{
int i,j,n,m,a,b,t;
scanf("%d",&t);
while(t--)
{
init();
scanf("%d%d",&n,&m);
for(i=;i<n;i++)
{
scanf("%d%d",&a,&b);
add(a,b,);
}
int temp=dfs();
//if(temp+1>ans)
// ans=temp+1;
for(i=;i<=m;i++)
{
scanf("%d",&a);
if(a<=ans+)
printf("%d\n",a-);
else
printf("%d\n",(a-ans-)*+ans);
}
}
return ;
}