hdu 4607(树的直径)

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4607

题解:给定一棵树，从树中的任意选一个顶点出发，遍历K个点的最短距离是多少？(每条边的长度为1)

算法分析：

首先如果k小于等于直径长度，那么答案为k−1;
如果k大于直径长度，设直径长度为r，那么答案为r−1+(k−r)*2;
树的直径：树上的最长简单路径;

代码:

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;
#define for0(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define LL long long
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define PI 3.1415926
#define N 300005

int vis[N],head[N];
int n,maxx,k;

struct node 
{
    int x,y;
    int next;
}a[N];

void dfs(int x,int len,int vis[])
{
    if(len>maxx){
        maxx=len;
        k=x;
    }
    for(int i=head[x]; i!=-1; i=a[i].next){
        if(!vis[a[i].y]){
            vis[a[i].y]=1;
            dfs(a[i].y,len+1,vis);
            vis[a[i].y]=0;
        }
    }
}

void init()
{
    for(int i=1; i<=n; i++){
        vis[i]=0;
        head[i]=-1;
    }
}

void get()
{
    int u,v,l=1;
    for(int i=1; i<n; i++){
        scanf("%d%d",&u,&v);
        a[l].x=u;
        a[l].y=v;
        a[l].next=head[u];
        head[u]=l++;

        a[l].x=v;
        a[l].y=u;
        a[l].next=head[v];
        head[v]=l++;
    }
}

int main()
{
    int t,m;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        init();
        get();

        maxx=-1;
        vis[1]=1;
        dfs(1,0,vis);
        for (int i=1; i<=n; i++)
            vis[i]=0;  
        vis[k]=1;
        dfs(k,0,vis);
        for (int i=0; i<m; i++)  
        {  
            scanf("%d",&k);  
            if(k<=maxx+1)  
            {  
                printf("%d
",k-1);  
            }  
            else  
            {  
                printf("%d
",maxx+(k-maxx-1)*2);  
            }  
        }  
    }
    return 0;
}