From 34486e6c452bb1486fb916fbc1c158be02c64d6c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Proch=C3=A1zka?= <david@prochazka.dev>
Date: Wed, 10 Feb 2021 14:14:06 +0100
Subject: [PATCH] FIX: moved condition to better place, based on the suggestion
 of the linter

---
 src/mhtree/MHTree.java | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/mhtree/MHTree.java b/src/mhtree/MHTree.java
index 27fe6e7..7633d8c 100644
--- a/src/mhtree/MHTree.java
+++ b/src/mhtree/MHTree.java
@@ -57,7 +57,7 @@ public class MHTree extends Algorithm implements Serializable {
         PriorityQueue<ObjectToNodeDistanceRank> queue = new PriorityQueue<>();
         queue.add(new ObjectToNodeDistanceRank(root, object));
 
-        while (!queue.isEmpty() && operation.getAnswerCount() < operation.getK()) {
+        while (!queue.isEmpty()) {
             Node currentNode = queue.poll().getNode();
 
             if (currentNode.isLeaf()) {
@@ -67,6 +67,9 @@ public class MHTree extends Algorithm implements Serializable {
 
                     operation.addToAnswer(obj);
                 }
+
+                if (operation.getAnswerCount() >= operation.getK())
+                    break;
             } else {
                 for (Node child : ((InternalNode) currentNode).getChildren())
                     queue.add(new ObjectToNodeDistanceRank(child, object));
-- 
GitLab