Fixed a few things in lca.cpp
This commit is contained in:
parent
fc0be86143
commit
dcb2b2d376
|
@ -7,12 +7,12 @@ namespace lca {
|
|||
}
|
||||
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)];
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue