Get Minimum Squares Leetcode Python Interview