diff --git a/src/main/java/rrcf/general/Tree.java b/src/main/java/rrcf/general/Tree.java index 98090db..db180ce 100644 --- a/src/main/java/rrcf/general/Tree.java +++ b/src/main/java/rrcf/general/Tree.java @@ -276,11 +276,11 @@ public Leaf insertPoint(double[] point, Object index) { for (int i = 0; i < size(); i++) { double[][] bbox = node.point; Cut c = insertPointCut(point, bbox); - if (c.value < bbox[0][c.dim]) { + if (c.value <= bbox[0][c.dim]) { leaf = new Leaf(point, i); branch = new Branch(c, leaf, node, leaf.num + node.num); break; - } else if (c.value >= bbox[bbox.length - 1][c.dim] && point[c.dim] > c.value) { + } else if (c.value >= bbox[bbox.length - 1][c.dim]) { leaf = new Leaf(point, i); branch = new Branch(c, node, leaf, leaf.num + node.num); break; @@ -571,4 +571,4 @@ public Cut(int d, double v) { value = v; } } -} \ No newline at end of file +}