以下是一个用Python编写的函数,可以根据输入的整数n生成一个螺旋格式的NxN二维数组:
def generate_spiral_matrix(n):
# 创建一个空的NxN二维数组
matrix = [[0] * n for _ in range(n)]
# 定义四个边界
top, bottom, left, right = 0, n-1, 0, n-1
# 定义当前要填充的数字
num = 1
while top <= bottom and left <= right:
# 从左到右填充上边界
for i in range(left, right+1):
matrix[top][i] = num
num += 1
top += 1
# 从上到下填充右边界
for i in range(top, bottom+1):
matrix[i][right] = num
num += 1
right -= 1
# 从右到左填充下边界
for i in range(right, left-1, -1):
matrix[bottom][i] = num
num += 1
bottom -= 1
# 从下到上填充左边界
for i in range(bottom, top-1, -1):
matrix[i][left] = num
num += 1
left += 1
return matrix
使用示例:
n = 4
spiral_matrix = generate_spiral_matrix(n)
for row in spiral_matrix:
print(row)
输出结果:
[1, 2, 3, 4]
[12, 13, 14, 5]
[11, 16, 15, 6]
[10, 9, 8, 7]
这个函数的时间复杂度为O(n^2),空间复杂度为O(n^2),其中n为输入的整数。
上一篇:编写一个函数,接受一个由两个单词组成的字符串,并返回True如果两个单词都以相同的字母开头。
下一篇:编写一个函数,接受一个整数k作为参数,检查一个整数列表的列表l,并返回列表中数值总和等于k的整数列表个数的计数。