hdu6031 Innumerable Ancestors

hdu6031 Innumerable Ancestors
倍增
题意
给定一张无向图，对于每组询问，给出两个集合 A 与 B
求 lca(x,y) 最深的时候的深度 x属于A y属于B

题解
首先 我们发现答案具有单调性，于是我们就可以二分这个答案 mid
然后 把 a 集合 的 这个深度的祖先 求出来 然后看 与b 集合的这个深度 祖先 是否有交集就行了
用set 判断 是否有交集
用倍增法 快速求出一个点的祖先 求一次 只要 log n

#include <bits/stdc++.h> 
#define ll long long  
#define cl(a,b) memset(a,b,sizeof(a)) 
using namespace std ;  

const int N = 1e5 + 11 ; 
struct Edge{
    int to,pre ; 
}e[N*2] ; 
int n,Q,cnt,k1,k2,ans,l,r ; 
int head[N],deep[N],anc[N][21],a[N],b[N] ;   

inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ; 
}

inline void add(int x,int y) 
{
    e[++cnt].to = y ; 
    e[cnt].pre = head[x] ; 
    head[x] = cnt ; 
}

inline void init()  
{
    cl( head,-1 ) ;
    cl( deep,0 ) ;  
    cl( anc,-1 ) ; 
    cnt = 0 ; 
}

inline void dfs(int u,int fa,int d) 
{
    deep[ u ] = d ; 
    for(int i=head[u]; ~i ; i = e[i].pre)
    {
        int v = e[ i ].to ; 
        if(v==fa) continue ; 
        anc[ v ][ 0 ] = u ; 
        dfs(v,u,d+1) ; 
    }
}

inline void lca_init() 
{
    for(int j=1; (1<<j) <= n ; j++) 
        for( int i=1;i<=n;i++ ) 
            if(anc[i][j-1]!=-1) 
                anc[i][j] = anc[anc[i][j-1]][j-1] ; 
}

inline int query(int u,int d) 
//  返回  u的 第 d 个父亲 
{
    if( d < 0 ) return -1 ; 
    if( d==0 ) return u ; 
    int i ; 
    for(i=0; (1<<i) <= d ;i++) ;   //  注意 这里有分号相当于 求 log  没有的话 死循环  
    i-- ; 
    for( ;i>=0;i--)
        if(d-(1<<i) >= 0) 
        {
            d-=(1<<i) ; 
            u = anc[u][i] ; 
        } 
    return u ; 
} 

inline bool check(int x) 
{
    set<int>s ; 
    for(int i=1;i<=k1;i++) 
    {
        int dis = deep[a[ i ]] - x ; 
        int ret = query(a[ i ],dis) ; 
        if(ret==-1) continue ; 
        s.insert(ret) ; 
    }
    for(int i=1;i<=k2;i++) 
    {
        int dis = deep[b[ i ]] - x ; 
        int ret = query( b[ i ],dis ) ; 
        if(s.count(ret)) return true ;   
    }
    return false ; 
}

int main() 
{ 
    while(~scanf("%d%d",&n,&Q)) 
    {
        init() ; 
        int x,y ; 
        for(int i=1;i<n;i++) 
        {
            scanf("%d%d",&x,&y) ; 
            add(x,y) ; add(y,x) ; 
        }
        dfs(1,1,1) ; 
        lca_init() ; 
        
        while(Q--) 
        {
            scanf("%d",&k1) ; 
            l = 1 ; r = 1 ; 
            for(int i=1;i<=k1;i++) 
                scanf("%d",&a[ i ]),r = max(r,deep[ a[ i ] ]) ; 
            scanf("%d",&k2) ; 
            for(int i=1;i<=k2;i++) 
                scanf("%d",&b[ i ]) ;  
            while(l<=r) 
            {
                int mid = ( l+r ) >>1 ; 
                if(check(mid)) 
                {
                    ans = mid ; 
                    l = mid + 1 ; 
                }
                else 
                    r = mid - 1 ; 
            } 
            printf("%d
",ans ) ; 
        }
    }
    
    return 0 ; 
}