HDU 4607 Park Visit【树的直径】

题意：有N个节点，M个询问，接着给出N-1条边，询问要访问k个点的最短距离

（转）一棵树，问从任意点出发，每次访问k个点走过的最少的边数

思路：我们可以这样考虑，假设我们走的起点为S，终点为T，那么路径S-T上的点我们必须都走到（因为是一棵树），那么如果路径S-T上的点大于或等于k，则我们只需要沿着路径一直走就行，所以最小长度为k-1，否则，我们一定是在路径上的一些点停下，然后转向其他分支然后再回来，此时才能满足一共走过k个点。假设从S到T的路径上一共有num个点，则最后我们要求的答案ans=num-1+？，这个？即为中间点（可以包括S和T）到别的分支再回来所要走的最小距离，因为每个分支互不影响，所以我们可以对每一个分支分开求，因为图为一棵树，所以我们即是求一棵子树从根节点到他的子节点中遍历x点再回来所要走的最短距离，我们可以有观察加数学归纳法证明这个值为2*x，即从一棵树的根结点往他的子树遍历x个点再回来的最小距离为2*x，所以？=2*（k-num），所以答案ans=num-1+（k-num）*2，要是这个值最小，则要使num最大，所以我们只要求得该树的直径即可，以下时代码，求树的直径是个很经典的问题了，

 

算法证明：（转）

树的直径(Diameter)是指树上的最长简单路。
直径的求法：两遍BFS (or DFS)
任选一点u为起点，对树进行BFS遍历，找出离u最远的点v
以v为起点，再进行BFS遍历，找出离v最远的点w。则v到w的路径长度即为树的直径
*简单证明
于是原问题可以在O(E)时间内求出

关键在于证明第一次遍历的正确性，也就是对于任意点u，距离它最远的点v一定是最长路的一端。
如果u在最长路上，那么v一定是最长路的一端。可以用反证法：假设v不是最长路的一端，则存在另一点v’使得(u→v’)是最长路的一部分，于是len(u→v’) > len(u→v)。但这与条件“v是距u最远的点”矛盾。
如果u不在最长路上，则u到其距最远点v的路与最长路一定有一交点c，且(c→v)与最长路的后半段重合(why?)，即v一定是最长路的一端

因为是树是连通的，所以u必有一条路径c和最长路径L相交，len(c)>=1，L被分为两部分，一部分l1，一部分l2
假设第一次dfs过后，所求最长路径lu端不在L上，那么len（lu）>=len(c)+len(l1)（l1，l2对称，取l1或者l2都一样）
len(l2+c+lu)>len(l1+l2),矛盾.

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
#define MAX 100005
#define ll __int64
#define FOR(n) for(int i=0;i<(n);i++)

vector<int>tree[MAX];
int d[MAX];
void init(int n)
{
    for(int i=1;i<=n;i++)
        tree[i].clear();
}
void dfs(int u,int f,int step)
{
    d[u] = step;
    int size = tree[u].size();
    for(int i=0;i<size;i++)
        if(tree[u][i] != f)
            dfs(tree[u][i],u,step+1);
}
int main()
{
    freopen("B.txt","r",stdin);
    int n,x,t,m,a,b,res,k;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init(n);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            tree[a].push_back(b);
            tree[b].push_back(a);
        }
        int res = 1;
        dfs(res,0,1);
        for(int i=1;i<=n;i++)
            if( d[i] > d[res])
                res = i;
        dfs(res,0,1);
        for(int i=1;i<=n;i++)
            if( d[i] > d[res])
                res = i;
        res = d[res];
        while(m--)
        {
            scanf("%d",&k);
            if(res >= k)
                printf("%d\n",k-1);
            else
                printf("%d\n",res-1+(k-res)*2);
        }
    }
    return 0;
}