Get Minimum Squares Leetcode Python Beginner