Fix java treap test

indy256 · Sep 1, 2024 · 88c6354 · 88c6354
1 parent ff3e348
commit 88c6354
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/java/structures/Treap.java b/java/structures/Treap.java
@@ -174,8 +174,15 @@ public static void main(String[] args) {
         treap = insert(treap, 3, 2);
         treap = insert(treap, 6, 1);
         System.out.println(kth(treap, 1).key);
-        System.out.println(query(treap, 1, 10).mx);
+
+        TreapAndResult treapAndResult1 = query(treap, 1, 10);
+        treap = treapAndResult1.treap;
+        System.out.println(treapAndResult1.mx);
+
         treap = remove(treap, 5);
-        System.out.println(query(treap, 1, 10).mx);
+
+        TreapAndResult treapAndResult2 = query(treap, 1, 10);
+        treap = treapAndResult2.treap;
+        System.out.println(treapAndResult2.mx);
     }
 }