Leetcode Coding Exercise On Numpy Array Size