hdu - 4607 - Park Visit

题意：N个城市形成一棵树，相邻城市之间的距离是1，问游K个城市的最短路程是多少，共有M次询问(1 <= N, M <= 100000, 1 <= K <= N)。

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

——>>把这棵树拉直，得最长路（直径）共有maxd个城市，如果K <= maxd，只要一直向前走，路程为K - 1；如果K > maxd，贪心策略为最长路一定要从头走到尾，中间补上缺少的城市，每补一个，距离+2，结果是maxd - 1 + (K - maxd) * 2。

对于最长路（直径），可用无根树转有根树，第一次任选一点作根转一次，然后找出最深的叶子，第二次以这个最深的叶子为根转一次，树的层数就是最长路的长度。

#include <cstdio>
#include <vector>

using namespace std;

const int maxn = 100000 + 10;
vector<int> G[maxn];
int d[maxn];

void dfs(int u, int fa, int cur) //现在的点为u，其父结点为fa，从根到u的经过cur个点
{
d[u] = cur;
int cnt = G[u].size();
for(int i = 0; i < cnt; i++)
{
int v = G[u][i];
if(v != fa) dfs(v, u, cur+1);
}
}

int main()
{
int T, N, M, u, v, i, j, K;
scanf("%d", &T);
while(T--)
{
scanf("%d%d", &N, &M);
for(i = 1; i <= N; i++) G[i].clear();
for(i = 0; i < N-1; i++)
{
scanf("%d%d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1, 0, 1); //第一次转有根树，找一个最长路的一端
j = 1;
for(i = 2; i <= N; i++) if(d[i] > d[j]) j = i;
dfs(j, 0, 1); //第二次转有根树，找最长路
j = 1;
for(i = 2; i <= N; i++) if(d[i] > d[j]) j = i;
int maxd = d[j], ret;
for(i = 0; i < M; i++)
{
scanf("%d", &K);
if(K <= maxd) ret = K - 1;
else ret = maxd - 1 + (K - maxd) * 2;
printf("%d\n", ret);
}
}
return 0;
}