Leetcode刷题记录 59. 螺旋矩阵 II

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

示例:

输入: 3
输出:
[
 [ 1, 2, 3 ],
 [ 8, 9, 4 ],
 [ 7, 6, 5 ]
]

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

思路：借用螺旋矩阵第一题的思维，总结其顺时针走势规律。

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        import numpy as np
        matrix=np.zeros([n,n],int)
        #处理特殊情况
        if(n==0):return matrix
        #保留本次循环中的四方位极值
        up=0
        down=n-1
        left=0
        right=n-1
        num=1
        while(True):
            for i in range(left,right+1):
                matrix[up][i]=num
                num+=1
            up+=1    
            if (up > down): break
            
            for i in range(up,down+1):
                matrix[i][right]=num
                num+=1
            right-=1
            if (right < left): break
                
            for i in range(right,left-1,-1):
                matrix[down][i]=num
                num+=1
            down-=1
            if (down < up): break   
                
            for i in range(down,up-1,-1):
                matrix[i][left]=num
                num+=1
            left+=1
            if (left > right): break
        return matrix

if __name__=="__main__":
    print(Solution().generateMatrix(5))