E' corretto quello che dici, fornendo un comparatore il treeMap usa appunto quest'ultimo per effettuare la comparazione degli oggetti.