DAY5_数组

螺旋矩阵II （59）

import numpy as np  #python
def totalFruit(n):
    row, col = n, n
    startx, starty = 0, 0
    matrix = np.zeros((row, col))
    count = 1
    loop,mid = n // 2, n // 2
    for offset in range(1,loop+1):
        for i in range(starty, n-offset):
            matrix[startx][i] = count
            count += 1
        for i in range(startx, n-offset):
            matrix[i][n-offset] = count
            count += 1
        for i in range(n-offset, starty, -1):
            matrix[n-offset][i] = count
            count += 1
        for i in range(n-offset, startx, -1):
            matrix[i][starty] = count
            count += 1
        startx += 1
        starty += 1
    if n % 2 == 1:
        matrix[mid][mid] = count
    return matrix

if __name__ == "__main__":
    n = 5
    print(totalFruit(n))

注意事项和错误记录

1. C++：一键注释 Ctrl + K +C,一键取消注释Ctrl + K +U。
2. 注意螺旋矩阵的边界

个人刷题记录整理，欢迎随时纠错和讨论!