螺旋矩阵,python实现

螺旋矩阵问题

给定一个n阶正方形矩阵,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。力扣原题

  • 这个问题不涉及什么算法问题,考察的就是个人对于代码的掌控和抽象。
    螺旋矩阵长的是这个样子
    螺旋矩阵,python实现_第1张图片
  • 处理这个问题就得提到二分法中关于边界条件的规定,早处理二分法问题的时候,有个区间开闭原则,即左闭右闭 [left,right]左闭右开 [left,right)
  • [left,right] 此时 while left<= right 因为left==right 是有意义的
  • [left,right) 此时 while left

进入正题

写螺旋数组问题,也是要保证始终按照一个区间去写,就是,不管是从左到右,还是从上到下,都要保证区间一直(左闭右开)

  • 填充上行从左到右
  • 填充右列从上到下
  • 填充下行从右到左
  • 填充左列从下到上
def function(n):
    matrix = [[0] * n for _ in range(n)]

    number = 1
    left, right, up, down = 0, n - 1, 0, n - 1
    while left < right and up < down:
        # 从左到右
        for i in range(left, right):
            matrix[up][i] = number
            number += 1

        # 从上到下
        for i in range(up, down):
            matrix[i][right] = number
            number += 1

        # 从右向左
        for i in range(right, left, -1):
            matrix[down][i] = number
            number += 1

        for i in range(down, up, -1):
            matrix[i][left] = number
            number += 1
        left += 1
        right -= 1
        up += 1
        down -= 1
    # n 为奇数的时候,正方形中间会有个单独的空格需要单独填充
    if n % 2 != 0: 
        matrix[n // 2][n // 2] = number
    return matrix

参考代码随想录

你可能感兴趣的:(Python,python,算法)