螺旋矩阵2.md

59.螺旋矩阵II

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

模拟顺时针顺序填充矩阵

模拟四个方向

• 从左向右：行索引不变，列索引增大
• 从上向下：行索引增大，列索引不变
• 从右向左：行索引不变，列索引减小
• 从下向上：行索引减小，列索引不变
• 初始化元素全部为0，当索引越界，或者碰到不为0的值，就应该改变方向

def generateMatrix(n):
    # 定义四个移动方向
    # (0, 1): 从左向右, row+0, col+1
    # (1, 0): 从上向下 row+1, col+0
    # (0, -1): 从右向左 row+0, col-1
    # (-1, 0): 从下向上 row-1, col+0
    dirs = [(0, 1), (1, 0), (0, -1), (-1, 0)]
    # 初始化
    matrix = [[0] * n for _ in range(n)]
    row, col, dirs_index = 0, 0, 0
    for i in range(1, n*n+1):
        matrix[row][col] = i
        # 计算下一个填充位置
        dx, dy = dirs[dirs_index]
        r = row + dx
        c = col + dy
        if r >= n or r < 0 or c >= n or c < 0 or matrix[r][c] > 0:
            # 改变方向
            dirs_index = (dirs_index + 1) % 4
            dx, dy = dirs[dirs_index]
        row = row + dx
        col = col + dy

    return matrix