LEETCODE | PYTHON | 59 | 螺旋矩阵Ⅱ

LEETCODE | PYTHON | 59 | 螺旋矩阵Ⅱ

1. 题目

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

示例 1：

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/first-letter-to-appear-twice
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 代码

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:

        #初始化
        left,right,up,down = 0,n-1,0,n-1
        res = [[0]*n for i in range(n)]
        num = 1

        #左开右闭式循环
        while left<=right and up<=down:

            #上端填充
            for i in range(left,right+1):
                res[up][i] = num
                num = num + 1
            up = up + 1

            #右端填充
            for i in range(up,down+1):
                res[i][right] = num
                num = num + 1
            
            right = right - 1
            
            #下端填充
            for i in range(right,left-1,-1):
                res[down][i] = num
                num = num + 1
            
            down = down - 1
            
            #左端填充
            for i in range(down,up-1,-1):
                res[i][left] = num
                num = num + 1
            
            left = left + 1
            
        
        return res