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

代码

#include <iostream>

using namespace std;

int q[110][110];

int main()
{
    
    int n, m;
    cin >> n >> m;
    
    
    int dx[] = {
    -1,0,1,0}, dy[] = {
    0,1,0,-1};
    int d = 1;
    
    int x = 0, y = 0;
    for(int i = 1; i <= n*m; i++)
    {
    
        q[x][y] = i;
        int ne_x = x + dx[d], ne_y = y + dy[d];
        if(ne_x < 0 || ne_x >= n || ne_y < 0 || ne_y >= m || q[ne_x][ne_y])
        {
    
            d = (d+1) % 4;
            ne_x = x+dx[d], ne_y = y+dy[d];
        }
        x = ne_x, y = ne_y;
    }
    
    for(int i = 0; i < n; i++)
    {
    
        for(int j = 0; j < m; j++)
            printf("%d ",q[i][j]);
        printf("\n");
    }
    
    
    return 0;
}