leetcode59——螺旋矩阵II

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

示例 1：

输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

示例 2：

输入：n = 1
输出：[[1]]

提示：

1 <= n <= 20

思路：这道题略难，能观察到是每个循环有四个步骤，

1. 行不变，列从左到右递增
2. 列不变，行从上到下递增
3. 行不变，列从右到左递减
4. 列不变，行从下到上递减

但对循环边界的处理不好把握，观察到每次循环后，行和列都会缩减2，所以迭代次数从1到n//2，表现在代码中就是:

loop = n//2;    
for offset in range(1, loop + 1)

而在大循环里的四个小循环中同样要处理好边界问题，用两个指针来控制，row，col从0开始，n-offset 控制尾巴，而首尾都会变化，每次大循环后，row，col都相加1，offset也会增加1

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        array = [[0] * n for _ in range(n)]
        num = 1
        loop, mid = n // 2, n // 2 
        row = 0
        col = 0
        for offset in range(1, loop + 1):
            
            for k in range(col, n - offset):
                array[row][k] = num
                num += 1

            for k in range(row, n - offset):
                array[k][n - offset] = num
                num += 1


            for k in range(n - offset, row , -1):
                array[n - offset][k] = num
                num += 1


            for k in range(n - offset, col, -1):
                array[k][col] = num
                num += 1

            row += 1
            col += 1
        
        if n % 2 != 0 :			# n为奇数时，填充中心点
            array[mid][mid] = num  
            
        return array 

上面代码中row 和col 其实可以统一成一个变量，假设为start，这样运行速度更快

class Solution:
    def generateMatrix(self, n: int) -> List[List[int]]:
        array = [[0] * n for _ in range(n)]
        num = 1
        loop, mid = n // 2, n // 2 
        start = 0
        for offset in range(1, loop + 1):
            
            for k in range(col, n - offset):
                array[start][k] = num
                num += 1

            for k in range(row, n - offset):
                array[k][n - offset] = num
                num += 1


            for k in range(n - offset, row , -1):
                array[n - offset][k] = num
                num += 1


            for k in range(n - offset, col, -1):
                array[k][start] = num
                num += 1

            start += 1
        
        if n % 2 != 0 :			# n为奇数时，填充中心点
            array[mid][mid] = num  
            
        return array