hdu 4607 Park Visit (求树的直径)

题意：

          给出树，求从任意点开始遍历 K 个节点最短路径长度是多少？

思路：

          对于求树的直径 

          dfs 一遍维护每个节点的最长子链和次长子链，同时两链相接维护最长链即可。

          对于所有小于树的直径的 K 个节点，每条边只需要走一遍。大于直径的其他节点所的对应的边要走两边。

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
using namespace std;
const int MAXN=1e5+100;
bool vis[MAXN];
int T,n,m,x,y;
int ans;
vector <int> mp[MAXN];
int dfs(int pos){
    vis[pos]=1;
    int len=mp[pos].size();
    int ans1=0,ans2=0,temp;
    for(int i=0;i<len;i++){
        if(!vis[mp[pos][i]]){
            temp=dfs(mp[pos][i]);
            if(temp>ans1){
                ans2=ans1;
                ans1=temp;
            }else{
                if(temp>ans2){
                    ans2=temp;
                }
            }
            ans=max(ans,ans1+ans2+1);
        }
    }
    return ans1+1;
}
int main()
{
    scanf("%d",&T);
    while(T--){
        memset(vis,0,sizeof(vis));
        scanf("%d%d",&n,&m);
        for(int i=1;i<n;i++){
            scanf("%d%d",&x,&y);
            mp[x].push_back(y);
            mp[y].push_back(x);
        }
        ans=0;
        dfs(1);
        int lim=ans;
        //cout<<"asd"<<ans<<endl;
        int k;
        while(m--){
            scanf("%d",&k);
            //cout<<"......................"<<endl;
            if(k<=lim)
                printf("%d\n",k-1);
            else
                printf("%d\n",lim+(k-lim)*2-1);
        }
        for(int i=1;i<=n;i++) mp[i].clear();
    }
}