2013年多校賽 hdu4607

題目傳送門

題意：有n個景點，存在n-1條雙向邊，長度爲1，遊玩k個景點，求最短路長

 

題解：這個圖符合樹的定義，要走最短的路，那麼就是要求回頭路最少

   那麼就考慮樹上最長距離，樹的直徑， 如果訪問的點大於樹的直徑就要走回頭路，答案即多的數乘2加上原來樹的直徑

   若k<直徑，只要走k-1長度即可

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
using namespace std;
vector<int> p[100005];
int v[100005];
int n,m;
int mmx,t;
void dfs(int u,int step)
{
	if(step>mmx)
	{
		t=u;
		mmx=step;
	}
	for(int i=0;i<p[u].size();i++)
	{
		if(!v[p[u][i]])
		{
			v[p[u][i]]=1;
			dfs(p[u][i],step+1);
			v[p[u][i]]=0;
		}
	}
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
	
	for(int i=0;i<100005;i++)
		p[i].clear();
	
	scanf("%d %d",&n,&m);
	
	for(int i=1;i<n;i++)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		p[a].push_back(b);
		p[b].push_back(a);
	}
	int ans,k;
	
	memset(v,0,sizeof(v));
	mmx=-1;
	v[1]=1;
	dfs(1,0);//遍歷後求出直徑一端 
	
	memset(v,0,sizeof(v));	
	mmx=-1;
	v[t]=1;//從這一段開始遍歷求直徑 
	dfs(t,0);//此時遍歷後mmx爲直徑 
	
	while(m--)
	{
		scanf("%d",&k);
		if(k<=mmx+1)
		{
			printf("%d\n",k-1);
		}else {
			printf("%d\n",(k-mmx-1)*2+mmx);
		}
	}
	
	
    }
	return 0;
}