hdu4607 Park Visit 树形DP版

题目大意：给一棵树，要经过K个不同的点，求走的路径最短是多少。每一条树上的边长度都是1.

先用DP求出树的直径len（表示直径上的边数）。

如果k<=len+1，答案为在直径上走k-1

如果k>len+1，答案为沿着直径走，遇到比较好的岔路口就进去再出来，因为不要求方案，所以默认她会走。。。 len + 2*（k-len-1）

在dog的帮助下理解了后面为什么这样算，学文化课大概真的会把脑子学坏。。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
using namespace std;
#define N 100010
int T, n, m, x,y,len,l,k;
struct node{
       int x,y,next;
}edge[2*N];
int head[N], f[N], g[N];
bool vis[N];
void add(int x, int y)
{
     l++;
     edge[l].x=x;edge[l].y=y;
     edge[l].next = head[x];
     head[x]=l;
}
void dfs(int x)
{
     vis[x]=1;
     for (int p = head[x];p;p=edge[p].next)
         if (!vis[edge[p].y])
         {
            int y=edge[p].y;
            dfs(y);
            if (f[y] + 1 > f[x]) g[x] = f[x],f[x] = f[y] + 1;
            else if (f[y]+1>g[x]) g[x]=f[y]+1;                    
         }
} 
int main()
{
    scanf("%d", &T);
    while (T--)
    {
          scanf("%d%d", &n, &m);
          l = 0;
          memset(head, 0, sizeof(head)); 
          memset(vis, 0, sizeof(vis));
          memset(f, 0, sizeof(f));
          memset(g, 0, sizeof(g));
          for (int i=1;i<=n-1;i++)
          {
              scanf("%d%d", &x, &y);
              add(x,y);
              add(y,x);
          }
          dfs(1);
          len = 0;
          for (int i=1;i<=n;i++)
              len = max(len, f[i]+g[i]);
          len++;
          for (int i=1;i<=m;i++)
          {
              scanf("%d", &k);
              if (k<=len) printf("%d\n", k-1); else printf("%d\n", len-1+2*(k-len));
          }
    }
}