aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAnthony Wang2021-01-21 20:18:42 -0600
committerAnthony Wang2021-01-21 20:18:42 -0600
commit9519fe7099eed3aec691b92d215b769cc1839e22 (patch)
tree683136baaa30f7fef972d9b37db50b0de8cd9dc4
parent6087e1adebc9025681c0efbaa59d530472af0d2f (diff)
Update lca.cpp
-rw-r--r--Graph/lca.cpp30
1 files changed, 13 insertions, 17 deletions
diff --git a/Graph/lca.cpp b/Graph/lca.cpp
index b6d695f..f45ee1c 100644
--- a/Graph/lca.cpp
+++ b/Graph/lca.cpp
@@ -1,18 +1,14 @@
-namespace lca {
- int d[MX], L[MX][20];
- void dfs(vector<int> * G, int u = 1, int p = 0) {
- d[u] = d[p] + 1, L[u][0] = p;
- for (int i = 0; i < 18 && L[u][i]; i++) L[u][i + 1] = L[L[u][i]][i];
- for (auto & v : G[u]) if (v != p) dfs(G, v, u);
- }
- int lca(vector<int> * G, int u, int v) {
- if (d[u] > d[v]) swap(u, v);
- for (int i = 18; i >= 0; --i) if (d[v] - (1 << i) >= d[u]) v = L[v][i];
- if (u == v) return u;
- for (int i = 18; 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];
- }
- inline int dist(vector<int> * G, int u, int v) {
- return d[u]+d[v]-2*d[lca(G, u, v)];
- }
+int d[MX], L[MX][20];
+void dfs(int u = 1, int p = 0) {
+ d[u] = d[p] + 1, L[u][0] = p;
+ for (int i = 0; i < 18 && L[u][i]; i++) L[u][i+1] = L[L[u][i]][i];
+ for (auto & v : G[u]) if (v != p) dfs(v, u);
}
+int lca(int u, int v) {
+ if (d[u] > d[v]) swap(u, v);
+ for (int i = 18; i >= 0; --i) if (d[v] - (1<<i) >= d[u]) v = L[v][i];
+ if (u == v) return u;
+ for (int i = 18; 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];
+}
+inline int dist(int u, int v) { return d[u]+d[v]-2*d[lca(u, v)]; }