LeetCode59. 螺旋矩阵 II（python）

59. 螺旋矩阵 II

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

示例:

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

解题思路：

对每个周期（排列一圈），每个方向排列（放置）元素的个数固定；

举例来说，对于n=4时，第一个周期每个方向都放置3（n-1）个元素，第二个周期每个方向都放置1（n-2-1）个元素；所以对n=n，第x个周期，每个方向放置(n-1-2*(x-1))个元素；

但是对于n为奇数的情况需要做特殊的处理，在开头就判断：n若为奇数，矩阵的中心位置matrix[(n-1)/2][(n-1)/2]直接等于n方，也就是需要放置的最后一个元素。

 

代码：

class Solution:
    def generateMatrix(self, n: int):
        if n == 1:
            return [[1]]
        matrix = [[1 for _ in range(n)] for _ in range(n)]
        
        if n%2 != 0:
            x = int((n-1)/2)
            matrix[x][x] = n**2
            
        x = 0
        y = 0
        num = 1
        
        for side in range(n, -1, -2):
            
            for i in range(y, y+side-1):
                matrix[x][i] = num
                num+=1
                y = i+1
            if num == n**2-1:
                break

            for i in range(x, x+side-1):
                matrix[i][y] = num
                num+=1
                x = i+1

            for i in range(y,y-side+1 , -1):
                matrix[x][i] = num
                num+=1
                y = i-1
                    
            for i in range(x, x-side+1, -1):
                matrix[i][y] = num
                num+=1
                x = i
            
            if num == n**2-1:
                break
            y+=1
                    
        return matrix