Library/Graph/lca.cpp

17 lines
491 B
C++
Raw Normal View History

2020-06-13 16:41:42 +00:00
int d[MAXN], L[MAXN][20];
vector<int> G[MAXN];
2019-08-01 21:53:59 +00:00
2020-06-13 16:03:55 +00:00
void dfs(int u = 1, int p = 0) {
2019-08-01 21:53:59 +00:00
d[u] = d[p] + 1;
2019-09-17 03:08:32 +00:00
L[u][0] = p;
for (int i = 0; i < 16 && L[u][i]; i++) L[u][i + 1] = L[L[u][i]][i];
for (auto& v : G[u]) if (v != p) dfs(v, u);
2019-08-01 21:53:59 +00:00
}
int lca(int u, int v) {
if (d[u] > d[v]) swap(u, v);
2019-09-17 03:08:32 +00:00
for (int i = 16; i >= 0; i--) if (d[v] - (1 << i) >= d[u]) v = L[v][i];
2019-08-01 21:53:59 +00:00
if (u == v) return u;
2019-09-17 03:08:32 +00:00
for (int i = 16; i >= 0; i--) if (L[u][i] && L[u][i] != L[v][i]) u = L[u][i], v = L[v][i];
return L[u][0];
}