hdu 4607 树的直径

思路：先求出树的直径，如果k比较小，则显然在直径上走是最优的。如果k比较多的话，意味着我们要走“往返路”去访问一些结点，则很显然最优解是使得走“往返路”访问的结点的数量最少，于是我们考虑在直径上走，不够的点我们通过走往返路去访问，这样可以使得不走往返路就访问的点（直径上的点）最多，也就是走往返路访问结点最少，是最优解。

树的直径的求法：先任取一点，求离该点最远的点u，则u一定为直径的两个端点之一，然后求出离u点最远的点，二者距离便是直径。

证明见：http://www.cnblogs.com/wuyiqi/archive/2012/04/08/2437424.html

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int N = 100001;
int head[N];
int n, m, e;
int node, dist;

struct Edge
{
    int v, next;
} edge[N * 2];

void addEdge( int u, int v )
{
    edge[e].v = v;
    edge[e].next = head[u];
    head[u] = e++;
}

void dfs( int u, int fa, int depth )
{
    if ( depth > dist )
    {
        dist = depth;
        node = u;
    }
    for ( int i = head[u]; i != -1; i = edge[i].next )
    {
        int v = edge[i].v;
        if ( v == fa ) continue;
        dfs( v, u, depth + 1 );
    }
}

int main ()
{
    int t;
    scanf("%d", &t);
    while ( t-- )
    {
        scanf("%d%d", &n, &m);
        e = 0;
        memset( head, -1, sizeof(head) );
        for ( int i = 1; i < n; i++ )
        {
            int u, v;
            scanf("%d%d", &u, &v);
            addEdge( u, v );
            addEdge( v, u );
        }
        dist = -1;
        dfs( 1, -1, 0 );
        dist = -1;
        dfs( node, -1, 0 );
        while ( m-- )
        {
            int k, ans;
            scanf("%d", &k);
            if ( k <= dist + 1 ) ans = k - 1;
            else ans = dist + ( k - dist - 1 ) * 2;
            printf("%d
", ans);
        }
    }
    return 0;
}