题目大意是给一颗树,n个点,1为根,m次询问,每次询问给出两个集合A,B。求集合A和集合B中的点的深度最大的LCA。
大致思路是深搜,处理出每个点的父节点,倍增法处理出anc数组。
每次询问时,二分目标深度,找出集合A中每个点在目标深度的祖先,并插入集合S,再找出集合B中每个点在目标深度的祖先,如果祖先在集合S中,说明这个深度A,B存在公共祖先。
T不大,sumK也不大,可以水过去。
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
#include<set>
#define mm(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=1e5+5;
const int maxe=2e5+50;
struct Node{
int v,next;
}edge[maxe];
int head[maxn],cnt;
int dep[maxn],fa[maxn];
int n,m;
int anc[maxn][20];
void init(){
mm(head,-1);cnt=0;
}
void addedge(int u,int v){
edge[cnt].v=v;
edge[cnt].next=head[u];
head[u]=cnt++;
}
void dfs(int u,int pre,int depth){
fa[u]=pre;
dep[u]=depth;
for(int k=head[u];k!=-1;k=edge[k].next){
Node& e=edge[k];
if(e.v==pre) continue;
dfs(e.v,u,depth+1);
}
}
int s1[maxn],s2[maxn];
void process(){
for(int i=0;i<n;i++){
anc[i][0]=fa[i];
for(int j=1;(1<<j)<n;j++)
anc[i][j]=-1;
}
for(int j=1;(1<<j)<n;j++){
for(int i=0;i<n;i++)
if(anc[i][j-1]!=-1){
int a=anc[i][j-1];
anc[i][j]=anc[a][j-1];
}
}
}
int query(int p,int depth){
if(depth<0) return -1;
if(depth==0) return p;
for(int i=20;i>=0;i--)
if(depth>=(1<<i)){
p=anc[p][i];
depth-=(1<<i);
}
return p;
}
int K1, K2;
bool check(int mid){
set<int> S;
for(int j=0;j<K1;j++){
int dis=dep[s1[j]]-mid;
int pre=query(s1[j], dis);
if(pre!=-1) S.insert(pre);
}
int flag=0;
for(int j=0;j<K2;j++){
int dis=dep[s2[j]]-mid;
int pre=query(s2[j], dis);
if(S.count(pre)) return true;
}
return false;
}
int main(){
int u, v;
while(~scanf("%d%d",&n,&m)){
init();
for(int i=1;i<n;i++){
scanf("%d%d",&u, &v);
u--, v--;
addedge(u,v);
addedge(v,u);
}
dfs(0,-1,1);
process();
for(int i=0;i<m;i++){
scanf("%d",&K1);
int L=1, R=1;
for(int j=0;j<K1;j++){
scanf("%d",&s1[j]);
s1[j]--;
R=max(R, dep[s1[j]]);
}
scanf("%d",&K2);
for(int j=0;j<K2;j++){
scanf("%d",&s2[j]);
s2[j]--;
}
while(L+1<R){
int mid=(L+R)>>1;
if(check(mid)) L=mid;
else R=mid-1;
}
printf("%d\n",check(R)?R:L);
}
}
return 0;
}