前言:很早之前学过一个基于并查集的LCA的离线算法,但这么久过去了,在比赛中没遇到过,反而是LCA倍增算法用的比较多。于是,这里就复习一下基于倍增的LCA在线算法。
正文:
在聊算法核心思想之前,我们先聊一聊一个常数优化:
vector<int>lg(n + 1);
for (int i = 1; i <= n; i++) {
lg[i] = lg[i - 1] + ((1 << lg[i - 1]) == i);
}
其中,关于这个常数优化可以自己动手推推。
LCA倍增的核心之一是预处理:
vector<int>dep(n + 1);
vector<vector<int>>pre(n + 1, vector<int>(25));
function<void(int, int)>dfs = [&](int now, int fa) {
pre[now][0] = fa; dep[now] = dep[fa] + 1;
for (int j = 1; j <= lg[dep[now]]; j++) {
pre[now][j] = pre[pre[now][j - 1]][j - 1];
}
for (int i = d[now]; i != -1; i = v[i].next) {
int to = v[i].v;
if (to == fa)continue;
dfs(to, now);
}
};
这里遵循了一个式子:
进而推出:
在这个预处理中,我初学的时候对下面这几行有点疑问:
for (int j = 1; j <= lg[dep[now]]; j++) {
pre[now][j] = pre[pre[now][j - 1]][j - 1];
}
在之前说明过:,那么这里为什么是
,实际上这个等于是不会影响的。加一个等于号无非是向上再多跑一层,如果没有那么长,它就会指向 0 。
LCA倍增核心之二是二进制思想:
for (int i = 1; i <= m; i++) {
int a, b;
cin >> a >> b;
if (dep[a] < dep[b])swap(a, b);//确保a的深度一定大于b的深度
while (dep[a] > dep[b]) {
a = pre[a][lg[dep[a] - dep[b]] - 1];
}
if (a == b) {
cout << b << '\n';
continue;
}
for (int i = lg[dep[a]] - 1; i >= 0; i--) {
if (pre[a][i] != pre[b][i]) {
a = pre[a][i];
b = pre[b][i];
}
}
cout << pre[a][0] << '\n';
}
其实理解这段代码的关键点是转化为二进制。我们先看第一个部分:将a和b转移到同一个高度:
while (dep[a] > dep[b]) {
a = pre[a][lg[dep[a] - dep[b]] - 1];
}
可以这样理解:
之前我们通过swap的操作保证a的深度一定大于b的深度,那么 (等于0的情况是不会进入while里面的,直接输出答案)我们假设
,再将len转化为二进制,例如len=5,则二进制下为101,因为
的作用就是找到二进制下最高位的下标,所以我们可以通过这种方式不断向上爬,这样可以保证最终a和b处于同一高度。
for (int i = lg[dep[a]] - 1; i >= 0; i--) {
if (pre[a][i] != pre[b][i]) {
a = pre[a][i];
b = pre[b][i];
}
}
这个操作的核心和上述一致。就自行理解吧!