Fixup tracked_samples to work properly with virtual_root

We should always have ``tree.num_tracked_samples(tree.virtual_root) == len(tracked_samples)``, but this isn't currently working in #1704

The tracked samples code all looks a bit dodgy to me, it needs a good looking at before we ship C 1.0.