蛇形矩阵

题目来源：AcWing 756. 蛇形矩阵
关键词：语法题、数组模拟

输入两个整数n和m，输出一个n行m列的矩阵，将数字 1 到 n * m 按照回字蛇形填充至矩阵中。

具体矩阵形式可参考样例。

输入格式
输入共一行，包含两个整数n和m。

输出格式
输出满足要求的矩阵。

矩阵占n行，每行包含m个空格隔开的整数。

数据范围
1≤n,m≤100
输入样例：

3 3

输出样例：

1 2 3
8 9 4
7 6 5

分析：

1. 用两个代表x、y方向的偏移量的数组，用来表示一个坐标的上下左右的位置的坐标
2. d表示当前移动的方向，上0，右1，下2， 右3。同时也对应偏移量数组中的坐标
3. 当前位置填入数，再看下一个位置是否“撞墙”（出界或者下一位置已经填过数），要是撞墙就顺时针旋转90度，然后更新当前坐标

c++实现：

#include <iostream>
using namespace std;

const int maxn = 110;
int n, m, ans[maxn][maxn];
int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1}; //4个方向偏移量

int main(){
  cin >> n >> m;
  
  int x = 0, y = 0, d = 1;  //xy当前坐标 d当前的方向
  for(int i = 1; i <= n * m; i ++ ){
      ans[x][y] = i;
      int a = x + dx[d], b = y + dy[d]; //下一个格子的坐标
      
      if(a < 0 || a >= n || b < 0 || b >= m || ans[a][b]){
          d = (d + 1) % 4;  //到下一个方向，顺指针转90度
          a = x + dx[d], b = y + dy[d]; 
      }
      x = a, y = b; //更新坐标
  } 
  
  for(int i = 0; i < n; i ++ ){
      for(int j = 0; j < m; j ++ ){
          cout << ans[i][j] << ' ';
          
      }
      cout << endl;
  }
    
    return 0;
}

---

python3实现：

# 读入n和m
n, m = map(int, input().split())

# ans: n行m列的结果数组 dx dy:偏移量数组
ans = [[0 for j in range(m)] for i in range(n)]
dx, dy = [-1, 0, 1, 0], [0, 1, 0, -1]

# xy当前坐标 d方向，也是偏移数组的下标
x, y, d = 0, 0, 1

for i in range(1, n * m + 1):
    ans[x][y] = i
    # a, b下一个位置
    a, b = x + dx[d], y + dy[d]
    # 如果越界了或者下一个数已经填过了
    if a < 0 or a >= n or b < 0 or b >= m or ans[a][b]:
        # 换方向
        d = (d + 1) % 4
        a, b = x + dx[d], y + dy[d]
    x, y = a, b
        
# 输出矩阵
for i in range(n):
    for j in range(m):
        print(ans[i][j], end=' ')
    print()