AcWing-754. 平方矩阵 II

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

数组的形式参照样例。

输入格式

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

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

输出格式

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

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

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

数据范围

0≤N≤100

输入样例：

1
2
3
4
5
0

输出样例：

1

1 2
2 1

1 2 3
2 1 2
3 2 1

1 2 3 4
2 1 2 3
3 2 1 2
4 3 2 1

1 2 3 4 5
2 1 2 3 4
3 2 1 2 3
4 3 2 1 2
5 4 3 2 1

#include<iostream>
using namespace std;

const int N = 110;
int arr[N][N];

int main() {
	int n;
	while (cin >> n, n) {
		int index = 0;
		int num = 1;
		// 从最中间斜向右下开始输入数据
		while (num <= n) {
			// 右上的一个斜下,最开始的元素位置在 arr[0][num-1]
			for (int i = 0, j = num - 1; i < n && j < n; i++, j++)arr[i][j] = num;
			// 左下的一个斜下，最开始的元素位置在 arr[num-1][0]
			for (int i = num - 1, j = 0; i < n && j < n; i++, j++)arr[i][j] = num;
			// num++;
			num++;
		}
		// 打印
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				cout << arr[i][j] << " ";
			}
			cout << endl;
		}
		cout << endl;
	}
	return 0;
}