牛客 - MMset2

传送门：牛客 - MMset2

思路分析

先找到点集中路径最长的两个点，那么答案一定是这条路径的中点到两端的距离（len/2）
如果总长是奇数的话，取大的那边，就是向上取整

样例输入

3
1 2
1 3
1
2 2 3

样例输出

1

AC代码

#include <functional>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <iomanip>
#include <vector>
#include <string>
#include <cstdio>
#include <chrono>
#include <random>
#include <queue>
#include <stack>
#include <cmath>
#include <map>
#include <set>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ls x<<1
#define rs x<<1|1
#define fi first
#define se second
#define ll long long
#define pb push_back
#define mp make_pair
#define fun function
#define sz(x) x.size()
#define lowbit(x) x&(-x)
#define all(x) x.begin(),x.end()
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);

using namespace std;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const int INF = 0x3f3f3f3f;
const int N=1e6+10;

vector<int>e[N];
int fa[N],dep[N],son[N],siz[N],top[N],a[N];

void dfs1(int u,int f) {
    fa[u]=f;
    dep[u]=dep[f]+1;
    siz[u]=1;
    int maxsize=-1;
    for(auto v:e[u]) {
        if(v==f)continue;
        dfs1(v,u);
        siz[u]+=siz[v];
        if(siz[v]>maxsize) {
            son[u]=v;
            maxsize=siz[v];
        }
    }
}

void dfs2(int u,int t) {
    top[u]=t;
    if(!son[u])
        return ;
    dfs2(son[u],t);
    for(auto v:e[u]) {
        if(v==son[u]||v==fa[u])
            continue;
        dfs2(v,v);
    }
}

int lca(int x,int y) {
    while(top[x]!=top[y]) {
        if(dep[top[x]]<dep[top[y]])    swap(x,y);
        x=fa[top[x]];
    }
    return dep[x]>dep[y]?y:x;
}

int dis(int u,int v){
    return dep[u]+dep[v]-2*dep[lca(u,v)];
}

int main() {
    IOS;
#ifdef xiaofan
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
#endif

    int n;
    cin>>n;
    for(int i=1;i<n;i++) {
        int u,v;
        cin>>u>>v;
        e[u].push_back(v);
        e[v].push_back(u);
    }
    int q;
    dfs1(1,1);
    dfs2(1,1);
    cin>>q;
    while(q--){
        int x;
        cin>>x;
        for(int i=1;i<=x;i++) cin>>a[i];
        int d=a[1];
        for(int i=2;i<=x;i++) if(dep[a[i]]>dep[d]) d=a[i];
        int ma=0;
        for(int i=1;i<=x;i++) ma=max(ma,dis(a[i],d));
        cout<<(ma+1)/2<<endl;
    }




    return 0;
}