【PAT乙级】螺旋矩阵

本题要求将给定的 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

思路

不停的画“同心圆”

CODE

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
bool cmp(int a, int b) {
	return a > b;
}
void judge(int** a, const int m,const int n,int begin,int end,const int *g,int position,int N) {
	if (position == N)
		return;
	a[begin][end] = g[position++];
	while (end+1<n&& a[begin][end + 1] == 0)
		a[begin][++end] = g[position++];
	while (begin+1<m&& a[begin + 1][end] == 0)
		a[++begin][end] = g[position++];
	while (end >0 && a[begin][end - 1] == 0)
		a[begin][--end] = g[position++];
	while (begin> 0&& a[begin - 1][end] == 0)
		a[--begin][end] = g[position++];
	return judge(a, m, n, begin , end + 1, g, position,N);
}
bool is_prime(int N)
{
	if (N == 1)return false;
	for (int i = 2; i <= sqrt(N); i++)
	{
		if (N% i == 0)
			return false;
	}
	return true;
}
int aab(int N) {
	int i = sqrt((double)N);//i是小于N的平方根并最接近N的平方根整数
	while (i >= 1) {
		if (N% i == 0)
			return i;
		i--;
	}
	return 1;
}
int main()
{
	int N;
	int m = 1, n = 1;
	cin >> N;
	n = aab(N);
	m = N / n;
	int** c = new int* [m];
	for (int i = 0; i < m; i++)
		c[i] = new int[n];
	for (int i = 0; i < m; i++)
		for (int j = 0; j < n; j++)
			c[i][j] = 0;
	int* g = new int[N];
	for (int i = 0; i < N; i++)
		scanf("%d",&g[i]);
	sort(g, g + N, cmp);
	if (is_prime(N))//如果是质数，那么直接输出
	{
		for (int i = 0; i < N; i++)
			printf("%d\n", g[i]);
		return 0;
	}
	judge(c, m, n, 0, 0, g, 0, N);
	for (int i = 0; i < m; i++)
	{
		for (int j = 0; j < n; j++)
			if (!j)
				printf("%d", c[i][j]);
			else
				printf(" %d", c[i][j]);
		printf("\n");
	}
}