螺旋矩阵问题
给定一个n阶正方形矩阵,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。力扣原题
- 这个问题不涉及什么算法问题,考察的就是个人对于代码的掌控和抽象。
螺旋矩阵长的是这个样子
- 处理这个问题就得提到二分法中关于边界条件的规定,早处理二分法问题的时候,有个区间开闭原则,即左闭右闭 [left,right] 和 左闭右开 [left,right)
- [left,right] 此时 while left<= right 因为left==right 是有意义的
- [left,right) 此时 while left<right
进入正题
写螺旋数组问题,也是要保证始终按照一个区间去写,就是,不管是从左到右,还是从上到下,都要保证区间一直(左闭右开)
- 填充上行从左到右
- 填充右列从上到下
- 填充下行从右到左
- 填充左列从下到上
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