Aula 8 [TreeSet e Comparator]

Oi Rafael,

só para eu entender, o que vc quis dizer com "ele não consegue buscar um dos funcionários com a idade igual"?

Abraços, Nico

Se você ver o resultado da execução, verificará que ele só vai exibir 2 nomes, porém foram adicionados 3 funcionários. Resolvi verificando se a idade é igual, e caso for, retorno -1. Mas gostaria de saber se isso é uma boa prática, ou se existe um outro meio de solucionar o problema.

Agora entendi! A saída ficou um pouco "tímida" na sua pergunta :)

Segue a resposta:

O TreeSet também é um Set e não permite elementos duplicados, com a diferença que o critério de igualdade é definido pelo seu Comparator (e não pelo hashCode e equals).

Para resolver o problema, dentro do Comparator, vc precisa levar em consideração o nome do aluno também, por exemplo:

public class OrdenaPorIdade implements Comparator<Funcionario> {

    @Override
    public int compare(Funcionario funcionario1, Funcionario funcionario2) {
        int diff = funcionario1.getIdade() - funcionario2.getIdade();

        if(diff == 0) {
            return funcionario1.getNome().compareTo(funcionario2.getNome());
        }

        return diff;
    }

}

e o teste:

public class Teste {
    public static void main(String args[]) {

        Funcionario f1 = new Funcionario("Barney", 13);
        Funcionario f2 = new Funcionario("Jonatan", 9);
        Funcionario f3 = new Funcionario("Guaraciara", 13);
        Funcionario f4 = new Funcionario("Guaraciara", 13);

        Set<Funcionario> funcionarios = new TreeSet<>(new OrdenaPorIdade());
        funcionarios.add(f1);
        funcionarios.add(f2);
        funcionarios.add(f3);
        funcionarios.add(f4);

        for (Funcionario funcionario : funcionarios) {
            System.out.println(funcionario.getNome());
        }

    }
}

Com a saída:

Jonatan
Barney
Guaraciara

Abs