Leetcode Coding Exercise On Numpy Matrix