HDOJ 4607 - Park Visit

用树形DP找到这颗树距离最远的两个点....对于一个询问K..若小于最远距离..显然只要走最远的这条边走K-1个边就行了..

当K大于了最远距离..就要去走其他的点...而去其他点必须再回到这个主线..所以要加上(最远距离上点总数-K)*2...

Program:

#include<iostream>
#include<stack>
#include<queue>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<cmath>
#define ll long long
#define oo 1000000007
#define MAXN 100010
using namespace std;
vector<int> Tree[MAXN];
int dis[MAXN],M;
void dfs(int x,int f)
{
int i,m=Tree[x].size();
dis[x]=1;
for (i=0;i<m;i++)
if (Tree[x][i]!=f)
{
dfs(Tree[x][i],x);
M=max(M,dis[x]+dis[Tree[x][i]]);
dis[x]=max(dis[x],dis[Tree[x][i]]+1);
}
return;
}
int main()
{
int T,i,n,k;
scanf("%d",&T);
while (~scanf("%d%d",&n,&k))
{
for (i=1;i<=n;i++) Tree[i].clear();
for (i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
Tree[x].push_back(y);
Tree[y].push_back(x);
}
memset(dis,0,sizeof(dis));
M=0;
dfs(1,0);
while (k--)
{
int x;
scanf("%d",&x);
if (x<=M) printf("%d\n",x-1);
else printf("%d\n",M-1+(x-M)*2);
}
}
return 0;
}