【PAT乙级】1050 螺旋矩阵 (25分)

本题要求将给定的 $N$ 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 $m$ 行 $n$ 列，满足条件：$m\times n$ 等于 $N$；$m\ge 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

解题思路：

答案代码：

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

const int maxn = 10005;
int a[maxn] = {0}, matrix[maxn][maxn];

bool cmp(int a, int b) {
	return a > b;
}

int main() {
	int N = 0;									// 数据存储 
	cin >> N;
	for (int i =0; i < N; i++)
		cin >> a[i];
	sort(a, a+N, cmp);	
    
    int m = sqrt(N * 1.0);						// m行 n列 
    int n = m;
    while (m * n != N) {
    	m++;	n = N/m;
	}
	
	int i = 0, cnt = 0;							// 螺旋存储 
	while (cnt < N) {
		for (int j=i; j<=n-1-i && cnt<N; j++)	matrix[i][j] = a[cnt++];
		for (int j=i+1; j<=m-1-i && cnt<N; j++) matrix[j][n-1-i] = a[cnt++];
		for (int j=n-2-i; j>=i && cnt<N; j--) 	matrix[m-1-i][j] = a[cnt++];
		for (int j=m-2-i; j>=i+1 && cnt<N; j--)	matrix[j][i] = a[cnt++];
		i++;
	}
	
	for (int i = 0; i < m; i++) {				// 矩阵打印 
		for (int j = 0; j < n; j++) {
			cout << matrix[i][j];
			if (j != n-1) cout << " ";
		}
		if (i != m-1) cout << endl;
	}
    
    return 0;
}