【ACWing】756. 蛇形矩阵

题目地址：

https://www.acwing.com/problem/content/758/

输入两个整数$n$和$m$，输出一个$n$行$m$列的矩阵，将数字$1$到$n\times m$按照回字蛇形填充至矩阵中。具体矩阵形式可参考样例。

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

输出格式：
输出满足要求的矩阵。矩阵占$n$行，每行包含$m$个空格隔开的整数。

数据范围：
$1\le n,m\le 100$

代码如下：

#include <iostream>
using namespace std;

int a[110][110];
int dx[] = {0, 1, 0, -1};
int dy[] = {1, 0, -1, 0};

int main() {
    int n, m;
    cin >> n >> m;
    for (int x = 0, y = 0, s = 1; s <= m * n; ) {
        for (int i = 0; i < 4; i++) {
            while (s <= m * n) {
                a[x][y] = s++;
                x += dx[i], y += dy[i];
                if (x < 0 || x >= n || y < 0 || y >= m || a[x][y]) {
                    x -= dx[i], y -= dy[i];
                    x += dx[(i + 1) % 4], y += dy[(i + 1) % 4];
                    break;
                }
            }
        }
    }

    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) printf("%d ", a[i][j]);
        puts("");
    }

    return 0;
}

时空复杂度$O(nm)$。