summaryrefslogtreecommitdiff
path: root/content/graph/LCA_sparse.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'content/graph/LCA_sparse.cpp')
-rw-r--r--content/graph/LCA_sparse.cpp32
1 files changed, 32 insertions, 0 deletions
diff --git a/content/graph/LCA_sparse.cpp b/content/graph/LCA_sparse.cpp
new file mode 100644
index 0000000..3e87cde
--- /dev/null
+++ b/content/graph/LCA_sparse.cpp
@@ -0,0 +1,32 @@
+struct LCA {
+ vector<ll> depth;
+ vector<int> visited, first;
+ int idx;
+ SparseTable st; //sparse table @\sourceref{datastructures/sparseTable.cpp}@
+
+ void init(vector<vector<int>>& adj, int root) {
+ depth.assign(2 * sz(adj), 0);
+ visited.assign(2 * sz(adj), -1);
+ first.assign(sz(adj), 2 * sz(adj));
+ idx = 0;
+ dfs(adj, root);
+ st.init(depth);
+ }
+
+ void dfs(vector<vector<int>>& adj, int v, ll d=0) {
+ visited[idx] = v, depth[idx] = d;
+ first[v] = min(idx, first[v]), idx++;
+
+ for (int u : adj[v]) {
+ if (first[u] == 2 * sz(adj)) {
+ dfs(adj, u, d + 1);
+ visited[idx] = v, depth[idx] = d, idx++;
+ }}}
+
+ int getLCA(int u, int v) {
+ if (first[u] > first[v]) swap(u, v);
+ return visited[st.queryIdempotent(first[u], first[v] + 1)];
+ }
+
+ ll getDepth(int v) {return depth[first[v]];}
+};