hdu6031 Innumerable Ancestors (二分+树上倍增)

题意

n个点的树，m个询问，每次询问
给出两个集合a和b，找到a的一个元素x，b的一个元素y，使得lca(x,y)深度最大
输出lca(x,y)的最大深度

思路：

深度满足二分性质：
深度大的如果满足，则深度小的也一定满足

二分深度，然后对于集合a和集合b的每个点，求他们的深度为mid的祖先，
如果深度为mid的祖先有交集说明可以找到x，y，lca(x,y)等于mid

求祖先用树上倍增

code:

#include<bits/stdc++.h>
using namespace std;
const int maxm=1e5+5;
const int maxd=16;
vector<int>g[maxm];
map<int,int>mark;
int a[maxm],b[maxm],aa,bb;
int f[maxm][20];
int d[maxm];
int n,m;
void dfs(int x,int fa){
    f[x][0]=fa;
    d[x]=d[fa]+1;
    for(int i=1;i<=maxd;i++){
        f[x][i]=f[f[x][i-1]][i-1];
    }
    for(int v:g[x]){
        if(v==fa)continue;
        dfs(v,x);
    }
}
int getfa(int x,int dep){//爬到dep层
    for(int i=maxd;i>=0;i--){
        if(d[f[x][i]]>=dep){
            x=f[x][i];
        }
    }
    return x;
}
bool check(int mid){//祖先的深度为mid
    mark.clear();
    for(int i=1;i<=aa;i++){
        if(d[a[i]]<mid)continue;
        mark[getfa(a[i],mid)]=1;
    }
    for(int i=1;i<=bb;i++){
        if(d[b[i]]<mid)continue;
        if(mark[getfa(b[i],mid)])return 1;
    }
    return 0;
}
int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        for(int i=1;i<=n;i++){
            g[i].clear();
        }
        for(int i=1;i<n;i++){
            int a,b;
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        dfs(1,0);
        while(m--){
            scanf("%d",&aa);
            for(int i=1;i<=aa;i++){
                scanf("%d",&a[i]);
            }
            scanf("%d",&bb);
            for(int i=1;i<=bb;i++){
                scanf("%d",&b[i]);
            }
            int ans=1;
            int l=1,r=n;
            while(l<=r){
                int mid=(l+r)/2;
                if(check(mid)){
                    ans=mid;
                    l=mid+1;
                }else{
                    r=mid-1;
                }
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}