AcWing 753. 平方矩阵 I 蛇形矩阵变形

输入整数 N，输出一个 N阶的回字形二维数组。

数组的最外层为 1，次外层为 2，以此类推。

输入格式

输入包含多行，每行包含一个整数 N。

当输入行为 N=0时，表示输入结束，且该行无需作任何处理。

输出格式

对于每个输入整数 N，输出一个满足要求的 N阶二维数组。

每个数组占 N行，每行包含 N个用空格隔开的整数。

每个数组输出完毕后，输出一个空行。

数据范围

0≤N≤100

输入样例：

1
2
3
4
5
0

输出样例：

1

1 1
1 1

1 1 1
1 2 1
1 1 1

1 1 1 1
1 2 2 1
1 2 2 1
1 1 1 1

1 1 1 1 1
1 2 2 2 1
1 2 3 2 1
1 2 2 2 1
1 1 1 1 1

#include<iostream>
using namespace std;

int a[100][100] = { 0 };
int main()
{
    int n;
    while(cin >> n, n)
    {
        for (int i = 0; i < n; i ++ )
        {
            for (int j = i; j < n - i; j ++ )
            {
                a[j][i] = a[i][j] = i + 1;
                a[j][n - i - 1] = a[n - i - 1][j] = i + 1;
            }
        }
        for (int i = 0; i < n; i ++ )
        {
            for (int j = 0; j < n; j ++ )
                cout << a[i][j] << ' ';
            cout << endl;
        }
        cout << endl;

    }
}

#include <iostream>
#include <cstring>
#include <cmath>

using namespace std;

const int N = 100 + 10;
int m[N][N];

int main(){
    int n;
    int dx[] = {-1, 0, 1, 0}, dy[] = {0, 1, 0, -1};

    while(cin >> n, n ){
        memset(m, 0, sizeof m);
        int d = 1, x = 0, y = 0;
        int cnt = 0;  // 表示改变方向次数
        int res = 1;  // 回形当前圈数
        for (int i = 0; i < n * n; i ++){
            int a = x + dx[d], b = y + dy[d];
            m[x][y] = res;

            if (a < 0 || a >= n || b < 0 || b >= n || m[a][b]){
                d = (d + 1) % 4;
                a = x + dx[d], b = y + dy[d];
                cnt ++;
                if (!(cnt % 4)) res ++;
            }
            x = a, y = b;
        }

        for (int i = 0; i < n; i ++){
            for (int j = 0; j < n; j ++)
                cout << m[i][j] << ' ';
            cout << endl;
        }
        cout << endl;
    }

    return 0;
}