树上找爸爸

倍增求lca

原理

我们定义fa[i][j] 是i向上走2^j 步的父亲，那么我们是不是可以通过dfs去预处理这个fa数组，具体就是fa[i][0] 肯定就是这个点的直系爸爸，我们在循环中每一次先向上跳 2^log(j-1) 在向上去跳一样的步子就可以求出fa[i][j]是谁了。

之后如何去找lca呢，我们采用倒叙去找爸爸的方法，如图
假如找6和4的lca，我们先将他们跳到同深度，之后先去找fa[4][log[deep]==fa[6][log[deep]]? 如果相同那么这个点一定在lca的上面，我们不去管他，假如不同我们就把x,y
跳到哪一个点上再去找，（注意循环不动，就是换x,y，要是动循环就有会有没用的操作），最后我们一定会跳到lca的两个不同的儿子上面，之后再去返回父亲的值就欧克了。

代码实现

预处理log[i]+1

llog[0]=1;
for(int i=1;i<=n;i++){
	llog[i]=llog[i-1]+(1<<llog[i-1]==i);
}

预处理dfs

void dfs(int x,int y){
	fa[x][0]=y;
	dep[x]=dep[y]+1;
	for(int i=1;(1<<i)<=dep[x];i++) fa[x][i]=fa[ fa[x][i-1] ][i-1];
	for(int i=head[x]i;i=nex[i]){
		if(to[i]!=y) dfs(to[i],x);
	}
}

找爸爸

int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);
	//默认x为最深的节点
	while(dep[x]>dep[y]) x=fa[x][ llog[dep[x]-dep[y]]+1];
	if(x==y) return x;
	for(int i=llog[dep[x]]-1;i>=0;i--){
		if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[x][y];
	}
	return fa[x][0];
}

完整代码

#include<bits/stdc++.h>
using namespace std;
const int MAXN=500005;
//倍增求lca，fa[i][j]	求的是从i开始向上走2^j步，找到的父亲
int head[2*MAXN],nex[2*MAXN],to[2*MAXN];//无向图记得开2倍 
int fa[MAXN][22];
int llog[MAXN];
int n,m,s;//节点	数，询问次数，根结点数 
int num=0;
void add(int x,int y){
	to[++num]=y;
	nex[num]=head[x];
	head[x]=num;
}



void get_input(){
	int x,y;
	scanf("%d%d%d",&n,&m,&s);
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);//这个题必须要加上双向边，因为我们还不知道谁是根节点 
	}
	llog[0]=1;
	for(int i=1;i<=n;i++){
		llog[i]=llog[i-1]+(1<<llog[i-1]==i);//预处理log2[i]+1的值 
	}
}
 
int dep[MAXN];

void dfs(int x,int y){
	dep[x]=dep[y]+1;
	fa[x][0]=y;
	for(int i=1;(1<<i)<=dep[x];i++){
		fa[x][i]=fa[ fa[x][i-1] ][i-1];
	}
	for(int i=head[x];i;i=nex[i]){
		if(to[i]!=y)dfs(to[i],x);
	}
}//dfs预先处理倍增数组 

int lca(int x,int y){
	if(dep[x]<dep[y]) swap(x,y);//假设x是最深的节点
	while(dep[x]>dep[y]) x=fa[x][ llog[ dep[x]-dep[y] ]-1 ]; 
	if(x==y) return x;
	for(int i=llog[dep[x]]-1;i>=0;i--){
		if(fa[x][i]!=fa[y][i]){
			x=fa[x][i];
			y=fa[y][i];
		}//如果不相等那么这个点一定不是lca，因为这是一颗树，所以lca上面的点一定是相等的
		 
	}
	return fa[x][0]; 
}

int main() {
	get_input();
	dfs(s,0);
	int q1,q2;
	while(m--){
		scanf("%d%d",&q1,&q2);
		printf("%d\n",lca(q1,q2));
	}
	return 0;
}

小结

比较简单的一个代码，但是在写的时候还是暴露出了不细心，无向图忘了开2倍数数组，等等都有问题，还是要再细心一点点啊。