https://www.luogu.org/problem/P3379
这次我是用树剖写的LCA,写的比倍增简单而且还快一点。
树剖一般指的是重链剖分,首先把所有节点的子节点中最大的标出来,称为重子节点,从节点到重子节点的边叫重边,一段连续的重边称为重链,其他的轻子节点叫轻链,然后把整棵树dfs一下,dfs的时候优先dfs重儿子,这样可以保证同一条重链上的点时间戳连续。这样就把一棵树剖分成了很多条链,那么就可以用数据结构维护这些链,优化时间复杂度。
如果询问两个点的LCA,先判断是不是在一条链上,如果不是那么要先比较这两个点的链头的深度,把深度大的点移到链头的父节点上,直到这两个点在一条链上,然后输出深度较小的点就好了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e5 + 10, mod = 1e9 + 7;
vector<int> g[maxn];
int fa[maxn],dep[maxn],sz[maxn],son[maxn],top[maxn];
//父亲、深度、大小、儿子、时间戳、重链顶、时间戳对应的编号、时间
void dfs1(int u, int p, int d)
{
fa[u] = p; dep[u] = d; sz[u] = 1;
int maxson = 0;
for(int v : g[u]){
if (v != p){
dfs1(v, u, d + 1);
sz[u] += sz[v];
if(sz[v] > maxson){
maxson = sz[v];
son[u] = v;
}
}
}
}
void dfs2(int u,int topf)
{
top[u] = topf;
// tid[u] = ++cnt;
// rnk[cnt] = u;
if (son[u]){
dfs2(son[u], topf);
for(int v : g[u]){
if(v != fa[u] && v != son[u]) dfs2(v, v);
}
}
}
int main()
{
int n, m, u, v, s;
scanf("%d %d %d", &n, &m, &s);
int t = n;
while (--t){
scanf("%d %d", &u, &v);
g[u].push_back(v);
g[v].push_back(u);
}
dfs1(s, s, 1);
dfs2(s, s);
while (m--){
scanf("%d %d", &u, &v);
while (top[u] != top[v]){
if (dep[top[u]] > dep[top[v]]) u = fa[top[u]];
else v = fa[top[v]];
}
printf("%d\n", dep[u] < dep[v] ? u : v);
}
return 0;
}