Hi @rolof,
Yes, the KD-tree can likely provide the desired functionality. Here is an example:
https://itk.org/ITKExamples/src/Numerics/Statistics/SpatialSearch/Documentation.html
More information can be found in Section 5.1.7 of the ITK Software Guide.