[模板]LCA最近公共祖先

最近公共祖先$LCA$

题目链接

$LCA$即求树上两点的最近公共祖先

\qquad 如何求树上两点的最近公共祖先？
\qquad 首先可以想到找到这两个点坐标，然后再向上找父节点，直至找到相同父节点
\qquad 但是这样做太慢了，会超时
怎么办呢?

倍增

\qquad 倍增，就是按照$2$的倍数来增大，也就是一次跳$1,2,4,8,16\cdots$但是我们不是按照从小到大的顺序跳，而是从大到小跳，即$16,8,4,2,1$为什么这样做呢？考虑$9$如果从小到大跳，$1+2+4+8$会大于$9$此时就需要回溯，直到回溯到$1+8$但是如果从大到小跳可以直接跳到$8+1$，显然省去很多不必要的回溯，也会将时间缩短为$O(nlogn)$

如何实现？

\qquad 用$deep[]$来存储每个点的深度，用$q[x][y]$表示第$x$个结点向上找$2^y$后的父亲
\qquad 这里需要用一个$dfs$来预处理每个节点的$deep$和$p$

void dfs(int x,int fa){
	deep[x] = deep[fa] + 1;
	q[x][0] = fa;
	for(int i = 1; 1 << i < deep[x]; i ++)
		q[x][i] = q[q[x][i - 1]][i - 1];		//核心，维护每个点初始的父亲
	for(int i = head[x]; i != -1; i = e[i].next)
		if(e[i].k != fa) dfs(e[i].k,x);
}

\qquad 这样的预处理后就可以进行$LCA$

int LCA(int x,int y){
	if(deep[x] > deep[y]) dwap(x,y);
	for(int i = 20; i >= 0; i --)							//先跳到同一层
		if(deep[x] <= deep[y] - (1 << i)) y = q[y][i];
	if(x == y) return x;
	for(int i = 20; i >= 0; i --)							//一起往上跳
		if(q[x][i] != q[y][i]){
			x = q[x][i];
			y = q[y][i];
		}
	return q[x][0];
}

附上$ACcode$

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int MAXN = 5e5 + 5;
int head[MAXN],deep[MAXN],p[MAXN][22];
int top = 0;

struct Node{
	int next,k;
}e[MAXN << 1];

inline int read(){								//快读模板
	int n = 0,l = 1;
	char c = getchar();
	while(c < '0' || c > '9'){
		if(c == '-') l = -1;
		c = getchar();
	}
	while(c >= '0' && c <= '9'){
		n = (n << 1) + (n << 3) + (c & 15);
		c = getchar();
	}
	return n * l;
}

void add(int x,int y){						//邻接链表
	e[top].k = y;
	e[top].next = head[x];
	head[x] = top++;
}

void dfs(int k,int fa){
	deep[k] = deep[fa] + 1;
	p[k][0] = fa;
	for(int i = 1;(1 << i) <= deep[k]; i ++)
		p[k][i] = p[p[k][i - 1]][i - 1];
	for(int i = head[k]; i != -1; i = e[i].next)
		if(e[i].k != fa) dfs(e[i].k,k);
}

int LCA(int x,int y){
	if(deep[x] > deep[y]) swap(x,y);
	for(int i = 20; i >= 0; i --)
		if(deep[x] <= deep[y] - (1 << i)) y = p[y][i];
	if(x == y) return x;
	for(int i = 20; i >= 0; i --)
		if(p[x][i] != p[y][i]){
			x = p[x][i];
			y = p[y][i];
		}
	return p[x][0];
}

int main(){
	memset(head,-1,sizeof(head));
	int n = read(),m = read(),s = read();
	for(int i = 1; i < n; i ++){
		int x = read(),y = read();
		add(x,y);
		add(y,x);
	}
	dfs(s,0);
	for(int i = 1; i <= m; i ++){
		int x = read(),y = read();
		printf("%d\n",LCA(x,y));
	}
	return 0;
}