最近公共祖先(LCA)之倍增法

什么是最近公共祖先？

LCA（Least Common Ancestors），即最近公共祖先，是指在有根树中，找出某两个结点u和v最近的公共祖先--------百度百科

例如：

 

 

对于点9和点5，它们的最近公共祖先是点2；对于点4和点2，它们的最近公共祖先是点2。

这里答案是显而易见，但是如果我们需要用暴力来计算的话，对于n个点，m次询问，时间复杂度为O(nm)，显然，当n和m过大的时候就TLE了，所以，我们需要一种更好的算法来降低时间复杂度，接下来的倍增法就是其中一种方法。

首先定义一个二维数组f,表示点i的祖先是fa[i][j]，然后定义一个数组depth，用来记录每一个点的深度。然后看到代码：

void dfs(int now, int fath) {//now表示当前节点，fath表示父亲节点。
    fa[now][0] = fath; depth[now] = depth[fath] + 1;//当且仅当now节点的祖先就是fath，深度是比父亲节点多一。
    for(int i = 1; i <= (int)(log(depth[now])/log(2))+1; ++i)//这里的lg就是以二为底，depth[now]的对数。
        fa[now][i] = fa[fa[now][i-1]][i-1];//当前now节点的祖先就是当前now节点的祖先的祖先。简单的来说就是。
    for(int i = head[now]; i; i = e[i].nex)//遍历当前now节点的所在的边，并且向下更新。
        if(e[i].t != fath) dfs(e[i].t, now);
}

这样我们就初始化所有的点的值了，时间复杂度为O(nlogn)。

然后就是求LCA，求LCA的代码如下：

int LCA(int x, int y) {
    if(depth[x] < depth[y]) swap(x, y);//求到底那一个最深。
    while(depth[x] > depth[y])
        x = fa[x][(int)(log(depth[x]-depth[y])/log(2))];//将其放到同一深度下。
    if(x == y) return x;//如果重合了的话，就返回x。
    for(int k = (int)(log(depth[x])/log(2)); k >= 0; --k)//这里一定要从后往前。原因下面会有解释。
        if(fa[x][k] != fa[y][k])//如果找到的祖先不一样，就继续向上找，如果一样的话，就往下找。
            x = fa[x][k], y = fa[y][k];
    return fa[x][0];//此时的父亲节点就是最近公共祖先。
}

对于上面为什么要从后往前找，比方说我们寻找7和11的最近公共祖先，如果我们从前往后找的话，我们会依次找到7->4->1,11->8->1,然后返回1，但是，1仅仅只是7,11的一个祖先，但不是最近的，如果我们从后往前找，可以直接精确找到此节点，具体操作可以自行体会。时间复杂度为O(mlogn)。

题目引入：祖孙询问

代码：

#include<bits/stdc++.h>
using namespace std;
struct zzz {
    int t, nex;
}e[500010 << 1]; 
int head[500010], tot;
void add(int x, int y) {
	e[++tot].t = y;
	e[tot].nex = head[x]; 
	head[x] = tot;
}
int depth[500001], fa[500001][22], lg[500001];
void dfs(int now, int fath) {
	fa[now][0] = fath; depth[now] = depth[fath] + 1;
	for(int i = 1; i <= (int)(log(depth[now])/log(2))+1; ++i)
		fa[now][i] = fa[fa[now][i-1]][i-1];
	for(int i = head[now]; i; i = e[i].nex)
		if(e[i].t != fath) dfs(e[i].t, now);
}
int LCA(int x, int y) {
	if(depth[x] < depth[y]) swap(x, y);
	while(depth[x] > depth[y])
		x = fa[x][(int)(log(depth[x]-depth[y])/log(2))];
	if(x == y) return x;
	for(int k = (int)(log(depth[x])/log(2)); k >= 0; --k)
		if(fa[x][k] != fa[y][k])
			x = fa[x][k], y = fa[y][k];
	return fa[x][0];
}
int main() {
	int n, m;
	scanf("%d", &n);
	int xx;
	for(int i = 1; i <= n; ++i) {
		int x, y; scanf("%d%d", &x, &y);
		if(y==-1){
			xx=x;
		}
		add(x, y); add(y, x);
	}
	dfs(xx, 0);
	scanf("%d",&m);
	for(int i = 1; i <= m; ++i) {
		int x, y, z; scanf("%d%d",&x, &y);
		if(LCA(x,y)==x){
			cout<<1<<endl;
		}
		else if(LCA(x,y)==y){
			cout<<2<<endl;
		}
		else {
			cout<<0<<endl;
		}
	} 
	return 0;
}