2019牛客暑期多校4A：meeting【树形dp】

题目：

2019牛客暑假多校4A：meeting

题意:

给定一颗树，有K个人在不同的节点，问最少需要多少时间他们就可以都在一个节点相遇

分析:

和HDU-2196有点相似，同样的做法，先随便找个根，DFS一遍预处理出以每个节点为根，其子树中的关键点到根的最长距离len1[]和次最长距离len2[]，然后换根dp，相当于枚举它们会在那个节点相遇；第二遍DFS处理出每个点经过其父节点的关键点的最远距离len3[]，如果len1[fa] = len1[son]+1，那么len3[son] = max(len2[fa]+1，len3[fa]+1)，注意len[] = 0就不能转移；K个人在节点x相遇的时间为max(len1[x],len3[x])；题解的做法是：两边dfs找到关键点的直径Len，Len/2就是答案

代码：

#include <bits/stdc++.h>
 
using namespace std;
const int maxn = 1e5+15;
int cnt,n,m,u,v,vis[maxn],head[maxn];
struct edge{
    int to,nxt;
}e[maxn<<1];
inline void add(int u,int v){
    e[++cnt] = (edge){v,head[u]};
    head[u] = cnt;
}
int len1[maxn],len2[maxn],len3[maxn];
void dfs1(int x,int fa){
    for(int i = head[x]; i;i = e[i].nxt){
        int v = e[i].to;
        if(v == fa) continue;
        dfs1(v,x);
        if(vis[v]){
            if(len1[x] < 1){
                len2[x] = len1[x];
                len1[x] = 1;
            }
            else if(len2[x] < 1) len2[x] = 1;
        }
        if(!len1[v]) continue;
        if(len1[x] < len1[v]+1){
            len2[x] = len1[x];
            len1[x] = len1[v]+1;
        }
        else if(len2[x] < len1[v]+1) len2[x] = len1[v]+1;
    }
}
void dfs2(int x,int fa){
    for(int i = head[x]; i; i=e[i].nxt){
        int v = e[i].to;
        if(v == fa) continue;
        if(len1[x] == len1[v]+1){
            len3[v] = max(len3[x],len2[x]);
            if(len3[v]) len3[v]++;
        }
        else{
            len3[v] = max(len3[x],len1[x]);
            if(len3[v]) len3[v]++;
        }
        dfs2(v,x);
    }
}
int main(){
    scanf("%d %d",&n,&m);
    for(int i = 1;i < n; ++i){
        scanf("%d %d",&u,&v);
        add(u,v); add(v,u);
    }
    for(int i = 1;i <= m; ++i) scanf("%d",&u),vis[u]++;
    dfs1(1,0); int ans = 1e9;
    dfs2(1,0);
    for(int i = 1;i <= n; ++i){
        ans = min(ans,max(len1[i],len3[i]));
    }
    printf("%lld",ans);
    return 0;
}