Fixed typo

cheran-senthil · Sep 25, 2023 · ec69341 · ec69341
1 parent 4d787b8
commit ec69341
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pyrival/graphs/binary_lifting_on_tree.py b/pyrival/graphs/binary_lifting_on_tree.py
@@ -99,11 +99,10 @@ def distance(self, a, b):
 
     def kth_ancestor(self, a, k):
         parent = self.parent
-        d = self.depth[a]
-        if d < k:
+        if self.depth[a] < k:
             return -1
-        for i in range(d.bit_length()):
-            if (d >> i) & 1:
+        for i in range(k.bit_length()):
+            if (k >> i) & 1:
                 a = parent[i][a]
         return a