Fixed a few things in lca.cpp

Changes

1 changed files (+3/-3)

• Graph/lca.cpp
  • @@ -7,12 +7,12 @@ 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]; + 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]; + 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*lca(G, u, v); + return d[u]+d[v]-2*d[lca(G, u, v)]; } }