From 88c635425d5797259efdf9e4cb9cb169103b6f28 Mon Sep 17 00:00:00 2001
From: Andrei Navumenka <indy256@gmail.com>
Date: Sun, 1 Sep 2024 14:12:56 -0400
Subject: [PATCH] Fix java treap test

---
 java/structures/Treap.java | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/java/structures/Treap.java b/java/structures/Treap.java
index f48603d6..3f8725e8 100644
--- 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);
     }
 }