1 parent e387311 commit 225ab92Copy full SHA for 225ab92
1 file changed
library/trees/ladder_decomposition/linear_kth_par.hpp
@@ -44,9 +44,9 @@ template<int KAPPA = 2> struct linear_kth_par {
44
}
45
int kth_par(int v, int k) {
46
assert(0 <= k && k <= d[v]);
47
- int j = v;
+ int anc_d = d[v] - k;
48
if (unsigned b = k / (KAPPA + 1); b)
49
- b = bit_floor(b), j = jmp[(pos[v] & -b) | b];
50
- return j = leaf[j], lad[j][k + d[j] - d[v]];
+ b = bit_floor(b), v = jmp[(pos[v] & -b) | b];
+ return v = leaf[v], lad[v][d[v] - anc_d];
51
52
};
0 commit comments