Count the Perfect Squares — medium Python for AI problem | Incognition