TreeSet
получает Comparator
, поэтому вы можете определить другой метод сравнения, например, я хочу отсортировать строки по второму символу
TreeSet<String> treeSet = new TreeSet<>(Comparator.comparing(a -> a.substring(1, 2)));
treeSet.add("avec");
treeSet.add("ipsum");
treeSet.add("lorem");
выведет
[lorem, ipsum, avec]