乙级 PAT 1050. 螺旋矩阵(25)

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行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

思路：用二维数组存储，按照层级进行，设置四个边界，一层层地填充，知道填充数字达到N为止。

    当N为完全二次数时，最里层是只有一个数的，此时需要特判输出。

#include<cstdio>
#include<algorithm>	
#include<cmath>
using namespace std;

int ma[10001][10001];//在主程序里定义会段错误

bool cmp(int a, int b) {
	return a > b;
}
int main() {
	int N, m, n;
	int a[10000];
	scanf("%d", &N);
	for (int i = 0; i < N; i++) {
		scanf("%d", &a[i]);
	}
	if (N == 1) {			//只有一个数直接输出
		printf("%d", a[0]);
		return 0;
	}

	sort(a, a + N, cmp);//从大到小排序

	for (n = sqrt(N); n > 0; n--) {		//求行m和列n
		if (N%n == 0) {
			m = N / n;
			break;
		}
	}
	int i = 1, j = 1, now = 0;	//i,j为当前要填的位置，now为序列带填的下标
	int U = 1, D = m, L = 1, R = n;		//4个边界
	while (now < N) {	//存到二维数组里面
		while (now < N&&j < R) {
			ma[i][j] = a[now++];
			j++;
		}
		while (now < N&&i < D) {
			ma[i][j] = a[now++];
			i++;
		}
		while (now < N&&j > L) {
			ma[i][j] = a[now++];
			j--;
		}
		while (now < N&&i > U) {
			ma[i][j] = a[now++];
			i--;
		}
		U++, D--, L++, R--;
		i++; j++;
		if (now == N - 1) {//最后一个数单独处理
			ma[i][j] = a[now++];
		}
	}
	for (i = 1; i <= m; i++) {	//打印
		for (j = 1; j <= n; j++) {
			printf("%d", ma[i][j]);
			if (j < n)	printf(" ");
			else	 printf("\n");
		}
	}

	return 0;
}