Python实现打印螺旋矩阵功能的方法

Python实现打印螺旋矩阵功能的方法

在编程中，打印螺旋矩阵是一个经典的算法问题。给定一个正整数N，我们需要生成一个N×N的矩阵，并按照顺时针螺旋的顺序填充从1到N²的数字。例如，当N=3时，矩阵如下所示：

1 2 3
8 9 4
7 6 5

当N=4时，矩阵如下所示：

1 2 3 4
12 13 14 5
11 16 15 6
10 9 8 7

传统的实现方法通常通过判断数据边界来实现，但这种方法的逻辑较为复杂，容易出错。因此，本文建议通过递归的方法来解决每一层数字的填充问题，使代码更加简洁和易读。

递归实现螺旋矩阵

递归方法的核心思想是逐层填充矩阵。每一层由四个方向（上、右、下、左）的循环组成，填充完一层后，递归处理下一层。具体实现步骤如下：

1. 初始化矩阵：创建一个N×N的二维数组，初始值为0。
2. 递归函数：定义一个递归函数dfs，参数包括矩阵arr、当前层的起始坐标(x, y)、当前起始数字start和当前层的边长n。
3. 填充每一层：根据当前层的边长n，分别向上、右、下、左四个方向填充数字。
4. 递归处理下一层：填充完当前层后，递归调用dfs函数处理下一层，起始坐标为(x+1, y+1)，起始数字为当前层的最后一个数字加1，边长减2。
5. 格式化输出：填充完成后，按照要求的格式打印矩阵。

代码实现

以下是完整的Python代码实现：

#coding:utf-8
n = int(raw_input('>'))
#初始化数组
arr = [[0]*n for i in range(n)]
#递归解决
def dfs(arr, x, y, start, n):
  if n<=0:return 0
  if n==1:
    arr[x][y] = start
    return 0
  #up
  for i in range(n):
    arr[x][y+i] = start
    start += 1
  #right
  for i in range(n-1):
    arr[x+1+i][y+n-1] = start
    start += 1
  #down
  for i in range(n-1):
    arr[x+n-1][y+n-2-i] = start
    start += 1
  #left
  for i in range(n-2):
    arr[x+n-2-i][y] = start
    start += 1
  dfs(arr,x+1,y+1,start,n-2)
a = dfs(arr,0,0,1,n)
#格式化输出print
l = len(str(n*n))+1
format = ('%'+str(l)+'d')*n
for tmp in arr:
  print format%tuple(tmp)

代码解析

输入处理：通过raw_input获取用户输入的整数N，并将其转换为整型。

矩阵初始化：创建一个N×N的二维数组arr，初始值为0。

递归函数dfs：

• 基准条件：如果当前层的边长n小于等于0，直接返回；如果n等于1，直接将起始数字赋值给当前坐标，并返回。
• 向上填充：从(x, y)开始，向右移动N步，依次填充数字。
• 向右填充：从(x+1, y+N-1)开始，向下移动N-1步，依次填充数字。
• 向下填充：从(x+N-1, y+N-2)开始，向左移动N-1步，依次填充数字。
• 向左填充：从(x+N-2, y)开始，向上移动N-2步，依次填充数字。
• 递归调用：填充完当前层后，递归调用dfs函数处理下一层。

格式化输出：根据N²的位数，确定每个数字的宽度，然后按照该宽度格式化输出矩阵。

示例运行

当输入N=3时，程序输出：

  1   2   3
  8   9   4
  7   6   5

当输入N=4时，程序输出：

  1   2   3   4
 12  13  14   5
 11  16  15   6
 10   9   8   7

总结

通过递归方法实现螺旋矩阵的打印，不仅代码简洁，而且逻辑清晰。这种方法避免了复杂的边界判断，提高了代码的可读性和可维护性。希望本文的介绍能帮助读者更好地理解和应用这一算法。