题解报告：hdu 4607 Park Visit（最长链）

Problem Description

Claire and her little friend, ykwd, are travelling in Shevchenko's Park! The park is beautiful - but large, indeed. N feature spots in the park are connected by exactly (N-1) undirected paths, and Claire is too tired to visit all of them. After consideration, she decides to visit only K spots among them. She takes out a map of the park, and luckily, finds that there're entrances at each feature spot! Claire wants to choose an entrance, and find a way of visit to minimize the distance she has to walk. For convenience, we can assume the length of all paths are 1.
Claire is too tired. Can you help her?

Input

An integer T(T≤20) will exist in the first line of input, indicating the number of test cases.
Each test case begins with two integers N and M(1≤N,M≤10⁵), which respectively denotes the number of nodes and queries.
The following (N-1) lines, each with a pair of integers (u,v), describe the tree edges.
The following M lines, each with an integer K(1≤K≤N), describe the queries.
The nodes are labeled from 1 to N.

Output

For each query, output the minimum walking distance, one per line.

Sample Input

1

4 2

3 2

1 2

4 2

2

4

Sample Output

1

4

解题思路：看到这题肯定会想到用树的直径来求解，题目要求从某一点出发，访问连续k个点走过的最少边数。显然这时如果k个点都在树的直径上，那么经过的边数一定是最少的且为k-1，否则(即k>maxdist)就会经过树直径外的一些点(有k-maxdist个)，并且剩下的这些点都被访问两次，而此时经过的边数最少为maxdist+(k-maxdist-1)*2。

AC代码一(499ms)：一次dfs。

#include<iostream>
#include<string.h>
#include<cstdio>
using namespace std;
const int maxn=1e5+5;
struct EDGE{int to,next;}edge[maxn<<1];
int t,n,m,k,x,y,cnt,res,maxdist,head[maxn];
void add_edge(int u,int v){
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
int dfs(int u,int fa,int &maxdist){
    int Dmax=0,Dsec=0;
    for(int i=head[u];~i;i=edge[i].next){
        int v=edge[i].to;
        if(v^fa){
            int nowd=dfs(v,u,maxdist)+1;
            if(nowd>Dmax)Dsec=Dmax,Dmax=nowd;
            else if(nowd>Dsec)Dsec=nowd;
        }
    }
    maxdist=max(maxdist,Dmax+Dsec);
    return Dmax;
}
int main(){
    while(~scanf("%d",&t)){
        while(t--){
            scanf("%d%d",&n,&m);
            memset(head,-1,sizeof(head));cnt=maxdist=0;
            while(--n){
                scanf("%d%d",&x,&y);
                add_edge(x,y);
                add_edge(y,x);
            }
            dfs(1,-1,maxdist);
            while(m--){
                scanf("%d",&k);
                if(k<=maxdist)printf("%d
",k-1);
                else printf("%d
",maxdist+(k-maxdist-1)*2);
            }
        }
    }
    return 0;
}

AC代码二(546ms)：两次bfs。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
struct EDGE{int to,next;}edge[maxn<<1];
struct node{
    int u,dep;
    node(int x,int y):u(x),dep(y){}
};
int t,n,m,x,y,k,cnt,maxdep,maxvex,head[maxn];bool vis[maxn];
queue<node> que;
void add_edge(int u,int v){
    edge[cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
}
void bfs(int u,int dep,int &maxdep,int &maxvex){
    while(!que.empty())que.pop();
    memset(vis,false,sizeof(vis));
    que.push(node(u,dep));vis[u]=true;
    while(!que.empty()){
        node nod=que.front();que.pop();
        for(int i=head[nod.u];~i;i=edge[i].next){
            int v=edge[i].to;
            if(!vis[v]){
                vis[v]=true;
                que.push(node(v,nod.dep+1));
            }
        }
        if(maxdep<nod.dep)maxdep=nod.dep,maxvex=nod.u;
    }
}
int main(){
    while(~scanf("%d",&t)){
        while(t--){
            scanf("%d%d",&n,&m);
            memset(head,-1,sizeof(head));cnt=maxdep=0;maxvex=1;
            while(--n){
                scanf("%d%d",&x,&y);
                add_edge(x,y);
                add_edge(y,x);
            }
            bfs(1,0,maxdep,maxvex);maxdep=0;
            bfs(maxvex,0,maxdep,maxvex);
            while(m--){
                scanf("%d",&k);
                if(k<=maxdep)printf("%d
",k-1);
                else printf("%d
",maxdep+(k-maxdep-1)*2);
            }
        }
    }
    return 0;
}