解题报告:首先如果 k 小于等于直径长度,那么答案为k−1。 如果 k 大于直径长度,设直径长度为 r,那么答案为r−1+(k−r)∗2。
先找树的最长路;
找树中任意一点,dfs找该点所能达到的最远的点vv,然后从vv点dfs找树的最长路。。
#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
int t,n,m;
vector<int> q[100010];
int dp[100010],deep[100010];
int vv;
int temp;
int dfs(int u,int fa)
{
int i,v;
if(q[u].size()==1&&u!=vv) return dp[u]=1;
for(i=0;i<q[u].size();i++){
v=q[u][i];
if(v!=fa){
int l=dfs(v,u);
if(l+1>dp[u])
dp[u]=l+1;
}
}
return dp[u];
}
void dfs1(int u)
{
for(int i=0;i<q[u].size();i++){
int v=q[u][i];
if(!deep[v]){
deep[v]=deep[u]+1;
dfs1(v);
}
}
}
int main()
{
int i,j,k;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
q[i].clear();
for(i=1;i<n;i++){
int a,b;
scanf("%d%d",&a,&b);
q[a].push_back(b);
q[b].push_back(a);
}
int mx=0;
memset(dp,0,sizeof(dp));
memset(deep,0,sizeof(deep));
deep[1]=1;
dfs1(1);
for(i=1;i<=n;i++)
if(deep[i]>mx){
mx=deep[i];
vv=i;
}
int r=dfs(vv,-1)-1;
while(m--){
int a;
scanf("%d",&a);
if(a<=r+1)
printf("%d\n",a-1);
else
printf("%d\n",(a-r-1)*2+r);
}
}
}