leetcode/draft.py

import numpy as np
n = 3
m = np.array(range(1, (n * n) + 1)).reshape(n, n)
print(m)