写模板, LCA。

基于有根树。

1 作用:可以找到离两个点最近的公共点,这个公共点也可以作为唯一路径的一个中间点。如果记录了两个点到这个公共点的一些数据,就可以处理一些问题。

2 实现方法:基于倍增的思想,对每个点记录了2的整数次方的距离条件下的每个祖先,可以通过递推来解决。

3 注意事项:最大深度应该是不超过n的一个以2为底的指数。在查找的时候每个点记录了一个深度,先让两个点深度一致,再去查找祖先。

void solve(){
int n, m, root;
cin >> n >> m >> root;

vector<vector> al(n + 1);
for (int i = 1; i < n; ++i){
int u, v;
cin >> u >> v;
al[u].emplace_back(v);
al[v].emplace_back(u);
}

int p = 0;
for (int i = 0; (1 << i) < n; ++i){
p = i;
}

vector depth(n + 1);
vector<vector> dp(n + 1, vector(p + 1));

function<void(int, int, int)> dfs = [&](int u, int p, int d){
depth[u] = d;
dp[u][0] = p;
for (int i = 1; (1 << i) <= d; ++i){
dp[u][i] = dp[dp[u][i - 1]][i - 1];
}
for (auto& v : al[u]){
if (v != p){
dfs(v, u, d + 1);
}
}
};

dfs(root, 0, 1);

while (m --){
int u, v;
cin >> u >> v;
if (depth[u] < depth[v]){swap(u, v);}
for (int i = p; i >= 0; --i){
if (depth[u] - (1 << i) >= depth[v]){
u = dp[u][i];
}
}
if (u == v){
cout << u << '\n';
continue;
}
for (int i = p; i >= 0; --i){
if (dp[u][i] != dp[v][i]){
u = dp[u][i];
v = dp[v][i];
}
}
cout << dp[u][0] << '\n';
}
}

posted @ 2024-03-28 09:53  _Yxc  阅读(38)  评论(0)    收藏  举报