LCA 树链剖分

刚打完LCA板子,写个东西记下

  • 树剖的LCA比倍增的快很多,代码又不长,还比较容易理解,当然树剖啦
  • 具体的什么轻链重链就不多赘述
  • 算法流程:
  1. dfs第一遍求出 结点i的深度, 以i为根的子树大小, 结点i的父亲, 并求出重链
  2. dfs第二遍求出 结点i所在重链的链顶(如果在重链上)
  3. 开始LCA,两个点往上找,深度大的点就往上跳,这个点如果在重链上,就跳到所在重链的链顶的父亲处,在轻链上,就直接跳到自己父亲上

代码:(洛谷LCA板子题

#include <bits/stdc++.h>
#define SIZE 500000
using namespace std;

int n, m, root, x, y, len;
int f[SIZE + 30], dep[SIZE + 30], siz[SIZE + 30], son[SIZE + 30], top[SIZE + 30], fa[SIZE + 30];

struct Edge {
	int s, next;
}e[SIZE * 4 + 30];
inline void AddEdge(int u,int v){
    e[++len]=(Edge){v, f[u]};
    f[u] = len;
}

inline void dfs1(int s) {//预处理dep, siz, fa, son
	siz[s] = 1;
	for (int i = f[s]; i; i = e[i].next) {
		int d = e[i].s;
		if (fa[s] != d) {
			fa[d] = s;
			dep[d] = dep[s] + 1;
			dfs1(d);
			siz[s] += siz[d];
			if (siz[son[s]] < siz[d])
				son[s] = d;
		}
	}
}
inline void dfs2(int s) {//预处理top 
	if (s == son[fa[s]])
		top[s] = top[fa[s]];
	else top[s] = s;
	for (int i = f[s]; i; i = e[i].next)
		if (e[i].s != fa[s])
			dfs2(e[i].s);
}

int LCA(int a, int b) {
	while (top[a] != top[b]) {
		if(dep[top[a]] < dep[top[b]]) a = fa[top[a]];
		else b = fa[top[b]];
	}
	if (dep[a] >= dep[b]) return b;
	else return a;
}

int main() {
	scanf ("%d%d%d", &n, &m, &root);
	for (int i = 1; i < n; ++i) {
		scanf ("%d%d", &x, &y);
		AddEdge(x, y); AddEdge(y, x);
	}
	dep[root] = 1;
	dfs1(root);
	dfs2(root);
	for (int i = 1; i <= m; ++i) {
		scanf ("%d%d", &x, &y);
		printf("%d\n", LCA(x, y));
	}
	return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值