python每日一题——19螺旋矩阵

题目

给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。

示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
示例 2:
输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]

答案

这道题要求我们按照顺时针螺旋顺序返回矩阵中的所有元素。我们可以使用四个变量来定义矩阵的上下左右边界,然后按照右-下-左-上的顺序遍历矩阵,同时更新这四个边界。

具体实现如下:

def spiralOrder(matrix):
    """
    :type matrix: List[List[int]]
    :rtype: List[int]
    """
    # 定义矩阵的上下左右边界
    top, bottom = 0, len(matrix) - 1
    left, right = 0, len(matrix[0]) - 1

    # 定义结果列表
    res = []

    # 按照右-下-左-上的顺序遍历矩阵
    while left <= right and top <= bottom:
        # 右
        for i in range(left, right + 1):
            res.append(matrix[top][i])
        top += 1

        # 下
        for i in range(top, bottom + 1):
            res.append(matrix[i][right])
        right -= 1

        # 左
        if top <= bottom:
            for i in range(right, left - 1, -1):
                res.append(matrix[bottom][i])
            bottom -= 1

        # 上
        if left <= right:
            for i in range(bottom, top - 1, -1):
                res.append(matrix[i][left])
            left += 1

    return res

在这个实现中,我们首先定义了矩阵的上下左右边界,然后按照右-下-左-上的顺序遍历矩阵,同时更新这四个边界。在遍历过程中,我们使用一个列表来保存遍历到的元素,最后返回这个列表即可。

你可能感兴趣的:(算法练习,python,矩阵,算法,leetcode)