最近公共祖先/LCA模板

惊觉自己还不会倍增。。。
做一发模板

1.倍增法

求LCA(最近公共祖先)算法流程:

  • 将深度更深的点跳到与另一个点同一高度
  • 两个点同时向上跳,直到找到LCA。

当然,我们可以用朴素算法,不过复杂度是O(n)的,会超时,这里我们用倍增来做。
倍增,简单来说就是“成倍增长”。
这里要用到一个定理:任何一个数都能表示为唯一的二进制分解形式。
显然,就像十进制下能不重复的表示数一样,二进制表示下的数也是唯一的。
因此,我们可以枚举二进制每一位上的数字模拟向上跳的过程,最终的数就是跳的步数,复杂度O(logN)。

关键点

我们从大到小枚举数位取0还是取1。
为什么从大到小呢?因为如果从大到小枚举,若这一位取1时,向上跳到了超过了LCA的点,那么这一位肯定只能取0。而从小到大枚举时,可能会出现这一位不用取的情况,就会“悔棋”。
用f[i][j]表示i的2^j级祖先 ,只需要一遍DFS转移就能求出来,在跳的时候很方便。

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 500001;

int fir[MAXN], nxt[MAXN << 1], to[MAXN << 1], cnt;
int dep[MAXN], f[MAXN][21]; //fa[i][j]表示i的2^j级祖先 

inline int read(){
    int k = 0, f = 1; char ch = getchar();
    while(ch < '0' || ch > '9'){if(ch == '-') f = -1; ch = getchar();}
    while(ch >= '0' && ch <= '9'){k = k*10 + ch - '0'; ch = getchar();}
    return k * f;
}

inline void add_edge(int a, int b){
	to[cnt] = b;
	nxt[cnt] = fir[a];
	fir[a] = cnt++;
}

void dfs(int u, int fa){ //u为当前节点,f为父亲节点
	dep[u] = dep[fa] + 1;
	f[u][0] = fa; //f是u的2^0级祖先(第一级祖先)
	for(int i = 1; (1 << i) <= dep[u]; i++)
		f[u][i] = f[f[u][i - 1]][i - 1];//u的2^i级祖先就是 u的2^(i - 1)级祖先的2^(i - 1)级祖先 //(2 ^ i = 2^(i - 1) + 2^(i - 1))
	
	for(int i = fir[u]; i != -1; i = nxt[i]){
		int v = to[i];
		if(v != fa)
			dfs(v, u);
	}
}

int LCA(int x, int y){
	//使x深度大于y的深度
	if(dep[x] < dep[y])
		swap(x, y);
	//把x提到和y同样深度 
	for(int i = 20; i >= 0; i--) //因为2的20次方在100w左右,所以倍增往上跳的时候不会跳超过20次 
		if(dep[y] <= dep[x] - (1 << i))
			x = f[x][i];
	//特判,若刚好跳到一样的点上,那这个点就是LCA
	if(x == y)
		return x;
	//x和y同时跳,跳到x,y的LCA的下面一层
	for(int i = 20; i >= 0; i--){
		if(f[x][i] == f[y][i])
			continue;
		else
			x = f[x][i], y = f[y][i];
	}
	return f[x][0]; 
}

int main(){
	freopen("in.txt", "r", stdin);
	memset(fir, -1, sizeof(fir));
	int n = read(), m = read(), s = read();
	for(int i = 1; i < n; i++){
		int x = read(), y = read();
		add_edge(x, y);
		add_edge(y, x);
	}
	
	dfs(s, 0);

	for(int i = 1; i <= m; i++){
		int a = read(), b = read();
		printf("%d\n", LCA(a, b));
	}
	return 0;
}

2.树链剖分

树链剖分的好处在于,空间小,常数小。当你把树链剖分写熟了之后还是挺简单的
大体地讲就是暴力沿着链向上跳,有点像修改路径的那个操作。
不会树链剖分的点这个

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 500001;

int fir[MAXN], nxt[MAXN << 1], to[MAXN << 1], cnt;
int dep[MAXN], fa[MAXN], size[MAXN], son[MAXN];
int top[MAXN];

void dfs1(int u, int f, int d){
	dep[u] = d, fa[u] = f, size[u] = 1, son[u] = 0;
	for(int i = fir[u]; i != -1; i = nxt[i]){
		int v = to[i];
		if(v == f) continue;
		dfs1(v, u, d + 1);
		size[u] += size[v];
		if(size[v] > size[son[u]]) son[u] = v;
	}
}

void dfs2(int u, int topf){
	top[u] = topf;
	if(!son[u]) return;
	dfs2(son[u], topf);
	for(int i = fir[u]; i != -1; i = nxt[i]){
		int v = to[i];
		if(v != fa[u] && v != son[u]) dfs2(v, v);
	}
}

int lca(int x, int y){
	while(top[x] != top[y]){
		if(dep[top[x]] < dep[top[y]]) swap(x, y);
		x = fa[top[x]];
	}
	if(dep[x] > dep[y]) swap(x, y);
	return x;
}

inline int read(){
	int k = 0, f = 1; char ch = getchar();
	while(ch < '0' || ch > '9'){if(ch == '-') f = -1; ch = getchar();}
	while(ch >= '0' && ch <= '9'){k = k*10 + ch - '0'; ch = getchar();}
	return k * f;
}

inline void add_edge(int a, int b){
	to[cnt] = b;
	nxt[cnt] = fir[a];
	fir[a] = cnt++;
}

int main(){
	memset(fir, -1, sizeof(fir));
	int n = read(), m = read(), s = read();
	for(int i = 1; i < n; i++){
		int a = read(), b = read();
		add_edge(a, b);
		add_edge(b, a);
	}
	dfs1(s, 0, 1); dfs2(s, s);
	for(int i = 1; i <= m; i++){
		int x = read(), y = read();
		printf("%d\n", lca(x, y));
	}
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值