There is the special case for Particles, since they have IDs where we can do some particle operations based on searching for their ID and not using the tree (which is obviously spatially sorted and therefore optimized for operations on space).

We could store a map of ID's to ParticleTreeElements and leverage the Map's optimizations for finding IDs.

This particular operation is certainly one that would benefit from that construct.