P3379 【模板】最近公共祖先（LCA）|倍增法练习

题目链接：

P3379 【模板】最近公共祖先（LCA） - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

好多题涉及到求公共祖先，所以学习一下倍增 法求最近公共祖先。

参考视频：

D09 倍增算法 P3379【模板】最近公共祖先（LCA）_哔哩哔哩_bilibili

 【算法】LCA&RMQ&树差分——保姆级教程-clint夜-稍后再看-哔哩哔哩视频 (bilibili.com)

 说明：

总结一下步骤：

1. 定义dep数组记录深度，edge二维数组存结点u的邻接点，fa[u][i]存结点u的i级祖先结点（i从0到19）
2. dfs建表：记录深度dep[u]，为父节点深度+1。fa[u][0]=father，0级祖先就是直接父节点，就可以得到每个结点的i级祖先（通过递推获得，u的i级祖先等于u的i-1级祖先的 i-1级祖先，相当于我一半一半的跳），所以级数从小到大枚举。两者计算完毕，向下遍历所有孩子结点，注意，要判断孩子是否为父节点，防止往回走。
3. lca计算公共祖先：首先，找到把u，v两点中深度更深的点记为u，u往上跳，直到深度等于v，每次往上跳先跳到跨级数大的祖先，减少跳的次数。代码对应：

	//使u跳到v的同一层 ，每次跨大一些的步数 
	for(int j=19;j>=0;j--){
		//大于或等于，可以往上跳 
		if(dep[fa[u][j]]>=dep[v]){
			u=fa[u][j];
		}
	}

    然后，跳到同一层后，特判，可能v是他的一个祖先，u会跳到v，那么v就是最近公共祖先 ，直接返回。

	//特判，可能v是他的一个祖先，会跳到v，v就是最近公共祖先 
	if(u==v) return v;

  如果不是，u，v同时往上跳， 每次跨大一些的步数 ，但是注意 ：因为每次跳的步数很大，所以跳到相同的结点只能保证是公共祖先，不能保证是最近的 ，故跳到的点不是公共祖先才跳上去，保证一直在最近祖先下面。直到不能往上跳为止，因为不能往上跳了，说明两个点在0级祖先（往上走一步，直接父节点）的下一层了，不然的话，它们至少还可以往上跳一步，跳到他们的直接父节点。此时可以保证这个直接父节点就是它们的最近公共祖先，所以返回此时的0级祖先。

	// 到达同一层，两个同时往上跳，每次跨大一些的步数 ，但是注意 ：因为每次跳的步数很大，
	//所以跳到相同的结点只能保证是公共祖先，不能保证是最近的 ，故跳到的点不是公共祖先才跳上去，保证一直在最近祖先下面
	for(int j=19;j>=0;j--){
		if(fa[u][j]!=fa[v][j]){
			u=fa[u][j],v=fa[v][j];
		}
	}

return fa[u][0];

代码

（注释把踩的坑备注了）：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e5+10;
//存边，双向，因为给的数据没有说明边的哪一点是父节点 
vector<int> edge[N];
//表示树的结点个数、询问的个数和树根结点的序号
int n,m,s;
//记录深度和多级祖先 
int dep[N];
int fa[N][21];
//dfs记录深度，祖先，向下遍历孩子 
void dfs(int u,int father){
	//记录深度，
	dep[u]=dep[father]+1;
	
	//通过递推计算祖先
	fa[u][0]=father;
	for(int i=1;i<=19;i++){
		fa[u][i]=fa[fa[u][i-1]][i-1];
	} 
	
	//向下遍历孩子 ，计算孩子的深度，祖先 
	for(int i=0;i<edge[u].size();i++){
		int son=edge[u][i];
		//双向的，要防止往回走 
		if(son!=father) dfs(son,u);
	}
	
}

int lca(int u,int v){
	//是交换u，v不是交换depu depv 
	if(dep[u]<dep[v]) swap(u,v);
	
	//使u跳到v的同一层 ，每次跨大一些的步数 
	for(int j=19;j>=0;j--){
		//大于或等于，可以往上跳 
		if(dep[fa[u][j]]>=dep[v]){
			u=fa[u][j];
		}
	}
	
	//特判，可能v是他的一个祖先，会跳到v，v就是最近公共祖先 
	if(u==v) return v;
	
	// 到达同一层，两个同时往上跳，每次跨大一些的步数 ，但是注意 ：因为每次跳的步数很大，
	//所以跳到相同的结点只能保证是公共祖先，不能保证是最近的 ，故跳到的点不是公共祖先才跳上去，保证一直在最近祖先下面
	for(int j=19;j>=0;j--){
		if(fa[u][j]!=fa[v][j]){
			u=fa[u][j],v=fa[v][j];
		}
	}
	
	return fa[u][0];
}

signed main(){
  
  	//关同步！！！
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	 
    cin>>n>>m>>s;
    for(int i=0;i<n-1;i++){
    	int x,y;
    	cin>>x>>y;
    	edge[x].push_back(y);
    	edge[y].push_back(x);
	}
	
	//从根节点出发，不一定是1 ，是s 
     dfs(s,0);
     
     while(m--){
     	int a,b;
     	cin>>a>>b;
     	cout<<lca(a,b)<<'\n';
	 }

  
  return 0;
}