willy108

hld lca :O

Mar 7th, 2022
1,194
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.81 KB | None | 0 0
  1.  
  2. int sz[MX], in[MX], out[MX], head[MX], dep[MX], par[MX], ind = 0;
  3.  
  4. void dfs(int x, int p){
  5.     sz[x] = 1;
  6.     dep[x] = dep[p] + 1;
  7.     par[x] = p;
  8.     for(auto &i : adj[x]){
  9.         if(i == p) continue;
  10.         dfs(i, x);
  11.         sz[x] += sz[i];
  12.         if(adj[x][0] == p || sz[i] > sz[adj[x][0]]) swap(adj[x][0], i);
  13.     }
  14. }
  15.  
  16. void dfs2(int x, int p){
  17.     in[x] = ind++;
  18.     for(auto &i : adj[x]){
  19.         if(i == p) continue;
  20.         head[i] = (i == adj[x][0] ? head[x] : i);
  21.         dfs2(i, x);
  22.     }
  23.     out[x] = ind;
  24. }
  25.  
  26. int lca(int a, int b){
  27.     while(head[a] != head[b]){
  28.         if(dep[head[a]] > dep[head[b]]) swap(a, b);
  29.         b = par[head[b]];
  30.     }
  31.     if(dep[a] > dep[b]) swap(a, b);
  32.     return a;
  33. }
  34.  
  35. int main(){
  36.    
  37.     dfs(1, 0);
  38.     head[1] = 1;
  39.     dfs2(1, 0);
  40.  
  41.     return 0;
  42. }
Advertisement
Add Comment
Please, Sign In to add comment