Numpy1 2149 Leetcode Problem