【LeetCode】顺时针打印矩阵(Numpy矩阵思路-“矩阵还是很好用的”)

题目:

【LeetCode】顺时针打印矩阵(Numpy矩阵思路-“矩阵还是很好用的”)_第1张图片

解题思路

这里提供一种思路,利用numpy库将列表转化为矩阵,一边遍历一边改变矩阵,实现顺时针的顺序依次打印出每一个数字。下面我们画个图解释一下:
【LeetCode】顺时针打印矩阵(Numpy矩阵思路-“矩阵还是很好用的”)_第2张图片
我们就是沿着箭头的方向遍历每一个值,通过观察遍历方向,我们可以发现,在遍历过程中,其过程是这样的:

  1. 第一行
  2. 最后一列
  3. 最后一行
  4. 第一列

那我们先遍历第一行[1, 2, 3, 4]接着删掉他们
【LeetCode】顺时针打印矩阵(Numpy矩阵思路-“矩阵还是很好用的”)_第3张图片
再遍历最后一列,删掉他们:
【LeetCode】顺时针打印矩阵(Numpy矩阵思路-“矩阵还是很好用的”)_第4张图片
接着跟着上面的过程遍历,直到矩阵里面没有值。

代码解释

我们先将matrix转换成array

    mat = np.array(matrix)

之后的思路就是我们一边访问,一边将访问到的行列删除掉,这样能保证最后的结果没有重复的值,并且每次访问的都是最后一行和最后一列。

res = list()
while True:
    res = res + mat[0, :].tolist()
    mat = np.delete(mat, 0, 0)
    if not rule(mat):
        break
    res = res + mat[:, -1].tolist()
    mat = np.delete(mat, -1, 1)
    if not rule(mat):
        break
    res = res + list(reversed(mat[-1, :].tolist()))
    mat = np.delete(mat, -1, 0)
    if not rule(mat):
        break
    res = res + list(reversed(mat[:, 0].tolist()))
    mat = np.delete(mat, 0, 1)
    if not rule(mat):
        break

在过程中我们需要对mat进行判断,当行数或者列数为0的时候,整个矩阵就遍历完了。

def rule(mat):
    m, n = mat.shape
    if m == 0 or n == 0:
        return False
    else:
        return True

完整代码:

def method(matrix):
    import numpy as np
    if len(matrix) == 0:
        return 0
    mat = np.array(matrix)
    res = list()
    while True:
        res = res + mat[0, :].tolist()
        mat = np.delete(mat, 0, 0)
        if not rule(mat):
            break
        res = res + mat[:, -1].tolist()
        mat = np.delete(mat, -1, 1)
        if not rule(mat):
            break
        res = res + list(reversed(mat[-1, :].tolist()))
        mat = np.delete(mat, -1, 0)
        if not rule(mat):
            break
        res = res + list(reversed(mat[:, 0].tolist()))
        mat = np.delete(mat, 0, 1)
        if not rule(mat):
            break

    return res

def rule(mat):
    m, n = mat.shape
    if m == 0 or n == 0:
        return False
    else:
        return True

if __name__ == '__main__':
    matrix = [[7],[9],[6]]
    res = method(matrix)
    print(res)

在这里只是提供一种思路,上面的代码有点臃肿,没有进行优化。

你可能感兴趣的:(LeetCode,python,leetcode,算法)