From 2fbca6db6587982b83205309d1d58c2ea58ab68c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Proch=C3=A1zka?= <david@prochazka.dev>
Date: Sat, 2 Jan 2021 10:05:16 +0100
Subject: [PATCH] ADD: search up to k in approxKNN

---
 src/mhtree/MHTree.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mhtree/MHTree.java b/src/mhtree/MHTree.java
index 213ecf1..ea98761 100644
--- a/src/mhtree/MHTree.java
+++ b/src/mhtree/MHTree.java
@@ -50,7 +50,7 @@ public class MHTree extends Algorithm implements Serializable {
         PriorityQueue<ObjectToNodeDistanceRank> queue = new PriorityQueue<>();
         queue.add(new ObjectToNodeDistanceRank(root, object));
 
-        while (!queue.isEmpty()) {
+        while (!queue.isEmpty() && operation.getAnswerCount() < operation.getK()) {
             Node currentNode = queue.poll().getNode();
 
             if (currentNode.isLeaf())
-- 
GitLab