社区讨论

10分求助

P3379【模板】最近公共祖先(LCA)参与者 4已保存回复 9

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
9 条
当前快照
1 份
快照标识符
@lo13o38y
此快照首次捕获于
2023/10/22 14:40
2 年前
此快照最后确认于
2023/11/02 14:11
2 年前
查看原帖
CPP
#include <cmath>
#include <iostream>
#include <vector>

using namespace std;

const int MaxN = 500010, MaxK = 20;

int n, m, s;
int f[MaxN][MaxK], d[MaxN];
vector<int> g[MaxN];

int Q(int u, int k) {
  for (int i = 0; i < MaxK; i++) {
    if (k & (1 << i)) {
      u = f[u][i];
    }
  }
  return u;
}

void DFS(int u, int fa, int cnt) {
  f[u][0] = fa;
  for (int i = 1; i < MaxK; i++) {
    f[u][i] = f[f[u][i - 1]][i - 1]; 
  }
  d[u] = cnt;
  for (int i : g[u]) {
    if (i == fa) continue;
    DFS(i, u, cnt + 1);
  }
}

int main() {
  ios::sync_with_stdio(0), cin.tie(0);
  cin >> n >> m >> s;
  for (int i = 1, u, v; i < n; i++) {
    cin >> u >> v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  DFS(s, s, 1);
  for (int x, y; m; m--) {
    cin >> x >> y;
    int md = min(d[x], d[y]);
    x = Q(x, d[x] - md), y = Q(y, d[y] - md);
    for (int i = 0; i < MaxK; i++) {
      if (x != y) {
        x = f[x][i], y = f[y][i];
      } 
    }
    cout << x << '\n';
  }
  return 0;
}

回复

9 条回复,欢迎继续交流。

正在加载回复...