[PAT乙级]1050 螺旋矩阵 (思路+精简代码)

1050 螺旋矩阵

本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列，满足条件：m×n 等于 N；m ≥ n；且 m−n 取所有可能值中的最小值。

输入格式：
输入在第 1 行中给出一个正整数 N，第 2 行给出 N 个待填充的正整数。所有数字不超过 10^​4​​ ，相邻数字以空格分隔。

输出格式：
输出螺旋矩阵。每行 n 个数字，共 m 行。相邻数字以 1 个空格分隔，行末不得有多余空格。

输入样例：

12
37 76 20 98 76 42 53 95 60 81 58 93

输出样例：

98 95 93
42 37 81
53 20 76
58 60 76

思路：经典题目，模拟的思想非常值得学习研究。接收输入数据和排序不在话下，关键在于如何螺旋式地填入数据。
想象自己动手画这样一个螺旋矩阵时，应该是先从第一行向右，再从最右侧的那一列向下，随后从底行向左，接着从第一列向上，每次转向都是因为遇到了边界或已填入的元素。
那么，我们也可以尝试模拟这样的过程。在计算出行列数后，创建一个四周扩大一圈的二维数组，将边界位置的值都设置为1，便于我们设置“墙”，然后从第一行开始向右，直到撞墙，撞墙则转向向下，持续到撞墙，以此类推，每次撞到墙或者填过的元素(也就是下一格的值不为0)，就转向。

下面给出一种可行的实现方式。

#include <stdio.h>
#include <math.h>
#include <algorithm>
int N, num[10001] = { 0 }, a[10002][102] = { 0 };	//二维数组开太大会溢出，行最多需要10002，列最多只需要102(测试点7段错误)
int main() {
	scanf("%d", &N);
	int n = sqrt(N);
	while (N % n) n--;
	int m = N / n, cnt = N, x = 1, y = 1;
	//int a[m + 2][n + 2] = { 0 };	//PAT允许用变量声明数组大小，VS不允许，如果这样做，上边全局声明要注释掉a[]
	for (int i = 0; i < n + 2; i++) a[0][i] = a[m + 1][i] = 1;	//填充边界，下同
	for (int j = 0; j < m + 2; j++) a[j][0] = a[j][n + 1] = 1;
	for (int i = 0; i < N; i++) scanf("%d", &num[i]);
	std::sort(num, num + N);	//默认升序排序，所以需要倒序访问
	while (cnt) {	//逐次写入，轮流向右向下向左向上，若不能再移动，直接填入最后一个数
		if (a[x - 1][y] && (a[x][y - 1] && !a[x][y + 1])) a[x][y++] = num[--cnt];	//向右，条件是上边有数且左边有数、右边没有数
																				//上边需要有数是防止第四个if向上时第一个if也成立
		else if (a[x - 1][y] && !a[x + 1][y]) a[x++][y] = num[--cnt];	//向下，条件是上边有数且下边没有数，其他类比
		else if (a[x][y + 1] && !a[x][y - 1]) a[x][y--] = num[--cnt];
		else if (a[x + 1][y] && !a[x - 1][y]) a[x--][y] = num[--cnt];
		else a[x][y] = num[--cnt];	//四周都已填满，但是还在while中，说明cnt=1，直接填入最后一个数即可
	}
	for (int i = 1; i < m + 1; i++)
		for (int j = 1; j < n + 1; j++) j == n ? printf("%d\n", a[i][j]) : printf("%d ", a[i][j]);
	return 0;
}