756. 蛇形矩阵

文章目录

• ​​Question​​
• ​​Ideas​​
• ​​Code​​

Question

输入两个整数 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

Ideas

Code

#include <iostream>
#include <cstdio>

using namespace std;

int q[100][100];

int main()
{
    int n,m;
    cin >> n >> m;
    
    int dx[4] = {0,1,0,-1};
    int dy[4] = {1,0,-1,0};
    
    // x,y：坐标 d：偏移量，k：待填数
    for (int x = 0,y = 0,d = 0,k = 1; k <= n*m; k++)
    {
        q[x][y] = k;
        
        // 下一个位置
        int a = x + dx[d];
        int b = y + dy[d];
        
        // 判断撞墙条件：1.越界 2.走过
        if (a < 0 || a >= n || b < 0 || b >= m || q[a][b])
        {
            d = (d + 1) % 4; // 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 << q[i][j] << ' ';
        cout << endl;
    }
}