hdu 4607 Park Visit

http://acm.hdu.edu.cn/showproblem.php?pid=4607

先求树的直径 方法：两遍bfs ，任选一点 a  求到a点最远的一点b ，然后 求到b点最远点 c

这样 b和c之间的路径为 树的直径

然后讨论

假设直径上的点个数为 k   要访问的点个数为 v

1）如果v<=k 则可以直接在直径上参观  路径为 v-1

2）如果v>k 首先还是要走直径 但是 m=v-k  这m个点要到其他枝叶上访问 一共要花费 2*m

代码：

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
#include<stack>
#include<vector>
#include<algorithm>
#include<queue>
#include<stdexcept>
#include<bitset>
#include<cassert>
#include<deque>
#include<numeric>

using namespace std;

typedef long long ll;
typedef unsigned int uint;
const double eps=1e-12;
const int INF=0x3f3f3f3f;
const ll MOD=1000000007;
const int N=200005;
int head[N],I;
struct node
{
    int j,next;
}edge[N];
int dist[N];
void add(int i,int j)
{
    edge[I].j=j;
    edge[I].next=head[i];
    head[i]=I++;
}
int bfs(int x1,int n)
{
    queue<int>qt;
    memset(dist,-1,sizeof(dist));
    dist[x1]=0;
    qt.push(x1);
    while(!qt.empty())
    {
        int x=qt.front();
        qt.pop();
        for(int t=head[x];t!=-1;t=edge[t].next)
        {
            int j=edge[t].j;
            if(dist[j]==-1)
            {
                dist[j]=dist[x]+1;
                qt.push(j);
            }
        }
    }
    int k=x1;
    for(int i=1;i<=n;++i)
    if(dist[i]>dist[k])
    k=i;
    return k;
}
int main()
{
    //freopen("data.in","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d %d",&n,&m);
        memset(head,-1,sizeof(head));I=0;
        for(int i=1;i<n;++i)
        {
            int l,r;
            scanf("%d %d",&l,&r);
            add(l,r);
            add(r,l);
        }
        int k=bfs(1,n);
        k=bfs(k,n);
        k=dist[k]+1;
        while(m--)
        {
            int a;
            scanf("%d",&a);
            if(a<=k)
            printf("%d
",a-1);
            else
            printf("%d
",k-1+2*(a-k));
        }
    }
    return 0;
}