KDTree

Example

object KDTreeSample {
  def main(args: Array[String]): Unit = {
    val points = Vector(
      Vector(2.0, 3.0),
      Vector(5.0, 4.0),
      Vector(9.0, 6.0),
      Vector(4.0, 7.0),
      Vector(8.0, 1.0),
      Vector(7.0, 2.0)
    )

    val tree = KDTree(points)

    val result1 = tree.contains(Vector(2.0, 3.0))
    val result2 = tree.range(Vector(3.0, 3.0), Vector(6.0, 8.0))
    val result3 = tree.nearest(Vector(5.0, 6.0))
    val result4 = tree.knn(Vector(5.0, 6.0), 3)

    println(s"result1: $result1")
    println(s"result2: $result2")
    println(s"result3: $result3")
    println(s"result4: $result4")
  }
}

Output

result1: true
result2: Vector(Vector(4.0, 7.0), Vector(5.0, 4.0))
result3: (2.0,Vector(4.0, 7.0))
result4: Vector((2.0,Vector(4.0, 7.0)), (4.0,Vector(5.0, 4.0)), (16.0,Vector(9.0, 6.0)))