Otimização da solução: como remover o ramo else e usar for/else?