If we write a specialized `merge_and_indices` function instead of using `merge_by_key` and a `counting_iterator`, we can improve the performance a bit.
If we write a specialized
merge_and_indicesfunction instead of usingmerge_by_keyand acounting_iterator, we can improve the performance a bit.