hdu 4607 Park Visit 多校第一场

求出树的最长链长度。。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

typedef long long lld;
const lld mod = 1000000007;
const int MAXN = 100010;
vector<int>vv[MAXN];
int dp[MAXN];
int mx;
void swap(int &a,int &b)
{
    if(a>b)
    {
        int t=a;
        a=b;
        b=t;
    }
}
void DP(int u,int p)
{
    dp[u]=1;
    int siz=vv[u].size();
    int m1=-1,m2=-1;
    for(int i=0;i<siz;i++)
    {
        int v=vv[u][i];
        if(v==p) continue;
        DP(v,u);
        dp[u]=max(dp[u],dp[v]+1);
        m2=max(dp[v],m2);
        swap(m2,m1);
    }
    mx=max(mx,dp[u]);
    mx=max(mx,m1+m2+1);
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) vv[i].clear();
        for(int i=1;i<n;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            vv[u].push_back(v);
            vv[v].push_back(u);
        }
        mx=0;
        DP(1,0);
        while(m--)
        {
            scanf("%d",&n);
            if(n<=mx)
                printf("%d\n",n-1);
            else printf("%d\n",2*n-mx-1);
        }
    }
}