Fix treetopologyex.11.cpp (#26)

* Update treetopologyex.11.cpp * Update treetopologyex.11.cpp
ivan100sic · Jun 14, 2024 · b957cf8 · b957cf8
1 parent 179561d
commit b957cf8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/trees/treetopologyex.11.cpp b/trees/treetopologyex.11.cpp
@@ -90,7 +90,7 @@ struct tree_topology_ex : tree_topology {
 
     int kth(int x, int y, int k) {
         int z = lca(x, y);
-        if (k <= d[x] - d[y]) {
+        if (k <= d[x] - d[z]) {
             return lift(x, k);
         } else {
             return lift(y, d[x] + d[y] - 2*d[z] - k);
@@ -100,7 +100,7 @@ struct tree_topology_ex : tree_topology {
 /*snippet-end*/
 
 int main() {
-    tree_topology_ex tt({{0, 1}, {1, 2}, {1, 3}}, 3);
+    tree_topology_ex tt({{0, 1}, {1, 2}, {1, 3}, {2, 4}, {4, 5}, {5, 6}, {3, 7}}, 3);
 
-    return tt.lca(2, 3) != 1 || tt.distance(2, 3) != 2;
+    return tt.lca(2, 3) != 1 || tt.distance(2, 3) != 2 || tt.kth(6, 7, 3) != 2;
 }