Update kSort.h

Add functionality to store indices in the tree and return the indices in order of how the vectors were sorted.