Use tree for key to node lookups instead of std::map.

[trquinn]

This may be a space saver, but there may be performance implications, so do some benchmarks before merging. This includes benchmarking the GPU code.