59.螺旋矩阵二 Python

题目：

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

示例:

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

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

思路：

螺旋的给矩阵赋值。每到边缘，变换遍历方向。

代码：

class Solution(object):
    def generateMatrix(self, n):
        """
        :type n: int
        :rtype: List[List[int]]
        """
        res=[[0 for i in range(n)] for j in range(n)]
        num=1
        i=0
        while num<=n*n:
            #从左向右罗列
            for k in range(i,n-i):
                res[i][k]=num
                
                num+=1
#从上到下遍历
            for k in range(i+1,n-i):
                res[k][n-i-1]=num
                num+=1
#从右往左遍历
            for k in range(n-i-2,i-1,-1):
                res[n-i-1][k]=num
                num+=1
#从下往上遍历
            for k in range(n-i-2,i,-1):
                res[k][i]=num
                num+=1
            
            i+=1
        return res