[已解决]如何输出蛇形矩阵

wp231957

FishC 发表于 2024-4-11 10:45
非常感谢您的指正和补充！您提到了八个方向，并给出了相应的情况划分，包括起点在左上、左下、右上、右 ...

嗯，谢谢，就是您这个代码的排版  真是一言难尽啊

FishC

wp231957 发表于 2024-04-11 10:46
嗯，谢谢，就是您这个代码的排版  真是一言难尽啊

对不起，我明白了！代码排版在确保易读性上确实有些欠缺。这里是经过改进的代码排版，希望这次更符合您的期望：

1. def generate_snake_matrix(size, start_point, direction):
   
2.     matrix = [[0 for _ in range(size)] for _ in range(size)]  # 创建一个空的矩阵
   
3.     x, y = start_point  # 获取起点坐标
   
4.     num = 1  # 要填充的数字
   
5.    
   
6.     while True:
   
7.         matrix[y][x] = num  # 填充当前位置的数字
   
8.         
   
9.         if direction == 'down':
   
10.             if y + 1 < size and matrix[y + 1][x] == 0:  # 向下移动
    
11.                 y += 1
    
12.             elif x + 1 < size and matrix[y][x + 1] == 0:  # 向右移动
    
13.                 x += 1
    
14.                 direction = 'right'
    
15.             else:
    
16.                 break
    
17.         elif direction == 'right':
    
18.             if x + 1 < size and matrix[y][x + 1] == 0:  # 向右移动
    
19.                 x += 1
    
20.             elif y - 1 >= 0 and matrix[y - 1][x] == 0:  # 向上移动
    
21.                 y -= 1
    
22.                 direction = 'up'
    
23.             else:
    
24.                 break
    
25.         elif direction == 'up':
    
26.             if y - 1 >= 0 and matrix[y - 1][x] == 0:  # 向上移动
    
27.                 y -= 1
    
28.             elif x - 1 >= 0 and matrix[y][x - 1] == 0:  # 向左移动
    
29.                 x -= 1
    
30.                 direction = 'left'
    
31.             else:
    
32.                 break
    
33.         elif direction == 'left':
    
34.             if x - 1 >= 0 and matrix[y][x - 1] == 0:  # 向左移动
    
35.                 x -= 1
    
36.             elif y + 1 < size and matrix[y + 1][x] == 0:  # 向下移动
    
37.                 y += 1
    
38.                 direction = 'down'
    
39.             else:
    
40.                 break
    
41.         
    
42.         num += 1
    
43.    
    
44.     return matrix
    
45. 
    
46. # 测试示例
    
47. size = 6  # 矩阵的大小
    
48. start_point = (0, 0)  # 起点坐标
    
49. direction = 'down'  # 初始方向
    
50. 
    
51. snake_matrix = generate_snake_matrix(size, start_point, direction)
    
52. 
    
53. # 输出蛇形矩阵
    
54. for row in snake_matrix:
    
55.     for num in row:
    
56.         print(str(num).rjust(2), end=' ')
    
57.     print()

复制代码

这次排版更加清晰易读，希望能让您满意！

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。