Python | Leetcode Python题解之第59题螺旋矩阵II

最新推荐文章于 2024-11-07 04:52:00 发布

原创最新推荐文章于 2024-11-07 04:52:00 发布 · 478 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Python #Leetcode #题解

分享专栏收录该内容

489 篇文章

订阅专栏

这段代码定义了一个名为`generateMatrix`的方法，用于生成一个nxn的矩阵，其中从左上角开始，按照数列顺序填充数字，然后按行和列对角线方向递减。这是一种常见的IT算法问题，常用于练习矩阵操作和递归思想。

题目：

题解：

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        matrix = [[0] * n for _ in range(n)]
        num = 1
        left, right, top, bottom = 0, n - 1, 0, n - 1

        while left <= right and top <= bottom:
            for col in range(left, right + 1):
                matrix[top][col] = num
                num += 1
            for row in range(top + 1, bottom + 1):
                matrix[row][right] = num
                num += 1
            if left < right and top < bottom:
                for col in range(right - 1, left, -1):
                    matrix[bottom][col] = num
                    num += 1
                for row in range(bottom, top, -1):
                    matrix[row][left] = num
                    num += 1
            left += 1
            right -= 1
            top += 1
            bottom -= 1

        return matrix