B - pog loves szh III
题目:添加链接描述
题意:找出区域l到r的LCA->找l和r的LCA
分析:
链式前向星存树,先用dfs处理结点倍增关系。
然后从循环处理较深结点,直至两节点位于同一层,即可共同向上找公共祖先:当两节点父节点相遇时,一定为最小的公共祖先。
具体见代码:
#include <iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=3e5+7;
int h[maxn],f[maxn][30],lg[maxn];
int head[maxn],nex[maxn<<1],to[maxn<<1],tot;
int n,m,s;
void add(int a,int b){//添加
to[++tot]=b;
nex[tot]=head[a];
head[a]=tot;
}//这辈子再也不用链式前向星
void dfs(int x,int fa){//dfs处理倍增
h[x]=h[fa]+1;f[x][0]=fa;//h表示层高
for(int i=1;(1<<i)<=h[x];i++)
f[x][i]=f[f[x][i-1]][i-1];//节点的2^i级祖先是结点的2^(i-1)级祖先的2^(i-1)级祖先
for(int i=head[x];i;i=nex[i]){
if(to[i]!=fa)dfs(to[i],x);
}
}
int LCA(int x,int y){
if(h[x]<h[y]) swap(x,y);//保证h[x]为大的那个
while(h[x]>h[y]) x=f[x][lg[h[x]-h[y]]-1];//使两结点为同一层
if(x==y) return x;//相遇则为所求
for(int i=lg[h[x]]-1;i>=0;i--){
if(f[x][i]!=f[y][i])
x=f[x][i],y=f[y][i];
}//向上查询直至父节点相遇
return f[x][0];//返回其父节点
}
int main(){
int n;
while(cin>>n){
memset(head,-1,sizeof(head));
tot=0;
for(int i=0;i<n-1;i++){
int u,v;scanf("%d%d",&u,&v);
add(u,v);
add(v,u);
}
dfs(1,0);
for(int i=1;i<=n;i++)lg[i]=lg[i-1]+(1<<lg[i-1]==i);//初始化log值
int q;
cin>>q;
while(q--) {
int x,y;scanf("%d%d",&x,&y);
printf("%d\n",LCA(x,y));
}
}
}