Essa foi minha solução, tem alguma coisa errada ou alguma lógica melhor que eu poderia aplicar?

Você está vendo a versão anterior da nova experiência da Alura que estamos preparando para você. Em breve, ela ganha uma identidade visual novinha totalmente pensada em potencializar seus estudos!
Essa foi minha solução, tem alguma coisa errada ou alguma lógica melhor que eu poderia aplicar?

Olá João! Tudo bem?
No geral, seu código está bem feito e atende perfeitamente aos requisitos dos desafios. Uma melhoria que você pode considerar é adicionar verificações para garantir que o usuário insira um número válido, especialmente nos prompts.
Bons estudos!