Problema do quadrado magico em Python