倍增LCA模板

Posted 2020-10-14

第二弹

namespace Tree {
#define travel(x, i) for (int i = fir[x]; i; i = e[i].nxt) 
  static const int N = 1e5 + 5;
  
  struct edge {
    int nxt, to;
  } e[N << 1];
  int fir[N], cnt;
  int dep[N], fa[N][18];
  
  inline void add(int x, int y) {
    e[++ cnt] = (edge){fir[x], y};
    fir[x] = cnt;
  }
  inline void Dfs(int x, int p) {
    fa[x][0] = p;
    dep[x] = dep[p] + 1;
    for (int i = 1; i < 18; i ++)
      fa[x][i] = fa[fa[x][i - 1]][i - 1];
    travel(x, i)
      if (e[i].to != p) Dfs(e[i].to, p);
  }
  inline int LCA(int x, int y) {
    if (dep[x] < dep[y]) swap(x, y);
    for (int i = 17; i && dep[x] != dep[y]; i --)
      if (dep[fa[x][i]] >= dep[y]) x = fa[x][i];
    if (x == y) return x;
    for (int i = 17; i && fa[x][0] != fa[y][0]; i --)
      if (fa[x][i] != fa[y][i]) {
        x = fa[x][i];
        y = fa[y][i];
      }
    return fa[x][0];
  }
}