AcWing756. 蛇形矩阵（蓝桥杯基础）

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

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

输入格式

输入共一行，包含两个整数 nn 和 mm。

输出格式

输出满足要求的矩阵。

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

数据范围

1≤n,m≤1001≤n,m≤100

输入样例：

3  3

输出样例：

1  2  3

8  9  4

7  6  5

代码如下：

#include<iostream>
using namespace std;

const int N = 110;
int n, m;
int res[N][N];

int main()
{
    cin >> n >> m;
    
    int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};
    int x = 0, y = 0, d = 1;
    
    for(int i = 1; i <= n * m; i ++)
    {
        res[x][y] = i;
        int a = x + dx[d], b = y + dy[d];
        if(a < 0 || a >= n || b < 0 || b >= m || res[a][b])
        {
            d = (d + 1) % 4;
            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 << res[i][j] << ' ';
       }
       cout << endl;
    }
    return 0;
}