hdu6031 Innumerable Ancestors

  1 #include <bits/stdc++.h> 
  2 #define ll long long  
  3 #define cl(a,b) memset(a,b,sizeof(a)) 
  4 using namespace std ;  
  5 
  6 const int N = 1e5 + 11 ; 
  7 struct Edge{
  8     int to,pre ; 
  9 }e[N*2] ; 
 10 int n,Q,cnt,k1,k2,ans,l,r ; 
 11 int head[N],deep[N],anc[N][21],a[N],b[N] ;   
 12 
 13 inline int read() 
 14 {
 15     int x = 0 , f = 1 ; 
 16     char ch = getchar() ; 
 17     while(ch<‘0‘||ch>‘9‘) { if(ch==‘-‘) f = -1 ; ch = getchar() ; } 
 18     while(ch>=‘0‘&&ch<=‘9‘) { x = x * 10+ch-48 ; ch = getchar() ; } 
 19     return x * f ; 
 20 }
 21 
 22 inline void add(int x,int y) 
 23 {
 24     e[++cnt].to = y ; 
 25     e[cnt].pre = head[x] ; 
 26     head[x] = cnt ; 
 27 }
 28 
 29 inline void init()  
 30 {
 31     cl( head,-1 ) ;
 32     cl( deep,0 ) ;  
 33     cl( anc,-1 ) ; 
 34     cnt = 0 ; 
 35 }
 36 
 37 inline void dfs(int u,int fa,int d) 
 38 {
 39     deep[ u ] = d ; 
 40     for(int i=head[u]; ~i ; i = e[i].pre)
 41     {
 42         int v = e[ i ].to ; 
 43         if(v==fa) continue ; 
 44         anc[ v ][ 0 ] = u ; 
 45         dfs(v,u,d+1) ; 
 46     }
 47 }
 48 
 49 inline void lca_init() 
 50 {
 51     for(int j=1; (1<<j) <= n ; j++) 
 52         for( int i=1;i<=n;i++ ) 
 53             if(anc[i][j-1]!=-1) 
 54                 anc[i][j] = anc[anc[i][j-1]][j-1] ; 
 55 }
 56 
 57 inline int query(int u,int d) 
 58 //  返回  u的 第 d 个父亲 
 59 {
 60     if( d < 0 ) return -1 ; 
 61     if( d==0 ) return u ; 
 62     int i ; 
 63     for(i=0; (1<<i) <= d ;i++) ;   //  注意 这里有分号相当于 求 log  没有的话 死循环  
 64     i-- ; 
 65     for( ;i>=0;i--)
 66         if(d-(1<<i) >= 0) 
 67         {
 68             d-=(1<<i) ; 
 69             u = anc[u][i] ; 
 70         } 
 71     return u ; 
 72 } 
 73 
 74 inline bool check(int x) 
 75 {
 76     set<int>s ; 
 77     for(int i=1;i<=k1;i++) 
 78     {
 79         int dis = deep[a[ i ]] - x ; 
 80         int ret = query(a[ i ],dis) ; 
 81         if(ret==-1) continue ; 
 82         s.insert(ret) ; 
 83     }
 84     for(int i=1;i<=k2;i++) 
 85     {
 86         int dis = deep[b[ i ]] - x ; 
 87         int ret = query( b[ i ],dis ) ; 
 88         if(s.count(ret)) return true ;   
 89     }
 90     return false ; 
 91 }
 92 
 93 int main() 
 94 { 
 95     while(~scanf("%d%d",&n,&Q)) 
 96     {
 97         init() ; 
 98         int x,y ; 
 99         for(int i=1;i<n;i++) 
100         {
101             scanf("%d%d",&x,&y) ; 
102             add(x,y) ; add(y,x) ; 
103         }
104         dfs(1,1,1) ; 
105         lca_init() ; 
106         
107         while(Q--) 
108         {
109             scanf("%d",&k1) ; 
110             l = 1 ; r = 1 ; 
111             for(int i=1;i<=k1;i++) 
112                 scanf("%d",&a[ i ]),r = max(r,deep[ a[ i ] ]) ; 
113             scanf("%d",&k2) ; 
114             for(int i=1;i<=k2;i++) 
115                 scanf("%d",&b[ i ]) ;  
116             while(l<=r) 
117             {
118                 int mid = ( l+r ) >>1 ; 
119                 if(check(mid)) 
120                 {
121                     ans = mid ; 
122                     l = mid + 1 ; 
123                 }
124                 else 
125                     r = mid - 1 ; 
126             } 
127             printf("%d\n",ans ) ; 
128         }
129     }
130     
131     return 0 ; 
132 }