Leetcode Coding Exercise On Numpy And Function