PAT (Basic Level) 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

解题思路：
对于本题的第一个点是在满足 m*n==N && m>n && m-n的值最小 的情况下求出m和n。
在解决m和n的问题之后，根据题意按矩阵顺时针非递增的顺序填到二维数组里，所以首先应该把给定的一维数组进行非递增排序。最后就是最难的“填空”。
对于每一个当前点，它的下一步的位置与当前点有四种关系，即 上、下、左、右。
我们把这种关系模拟到代码中就是 用dx数组标识横坐标，dy数组表示纵坐标。写出四种方向如下：

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

然后在循环中判断下一个位置是否“出界”，如果碰到出界，就要“转向”（只需要在每次转向时改变一下方向即可，即把dx dy的下标分别加1）。 知道填满N个数为止。最后正常输出。

附：本题对空间复杂度有要求，二维数组不要开得过大，否则段错误。由于m>n,所以第二维的大小是第一位的开平方即可。

代码：

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
int res[10000][100], flag[10000][100];
int cmp1(int x, int y)
{
	return x > y;
}
int main()
{
	int N, a[10000], min = 9999, m, n;
	cin >> N;
	for (int i = 0; i < N; i++)
		cin >> a[i];
	sort(a, a + N, cmp1);
	for (n = sqrt(N); n >= 1; n--)
	{
		if (N % n == 0) 
		{
			m = N / n;
			break;
		}
	}
	int dx[4] = { -1,0,1,0 }, dy[4] = { 0,1,0,-1 };
	for (int x = 0, y = 0, d = 1, k = 0; k < N; k++)
	{
		res[x][y] = a[k];
		flag[x][y] = 1;
		int a = x + dx[d], b = y + dy[d];
		if (a < 0 || a >= m || b < 0 || b >= n || flag[a][b])
		{
			d = (d + 1) % 4;
			a = x + dx[d], b = y + dy[d];
		}
		x = a, y = b;
	}
	for (int i = 0; i < m; i++)
	{
		for (int j = 0; j < n; j++)
		{
			cout << res[i][j] << (j == n - 1 ? "\n" : " ");
		}
	}
	return 0;
}

ps：博主能力有限，如果读者发现什么问题，欢迎私信或评论指出不足。欢迎读者询问问题，乐意尽我所能解答读者的问题。欢迎评论，欢迎交流。谢谢大家！