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

 

分析：

既然m-n取最小值，m就直接从N的平方根开始判断。创建一个二维数组。将输入的数组按递减排序后，依次以j递增，i递增，j递减，i递减将数传入二维数组对应位置。

 

代码：

#include<iostream>
#include<vector>
#include<algorithm>
#include<math.h>
using namespace std;

int main()
{
	int i, j, k, n, col, row;
	scanf("%d", &n);
	vector<int> v1(n);
	for (i = 0; i < n; i++)
		scanf("%d", &v1[i]);
	sort(v1.begin(), v1.end());
	for (row = ceil(sqrt(n)); row <= n; row++)
	{
		for (col = 1; col <= row && row*col != n; col++);
		if (row*col == n) break;
	}
	vector<vector<int>> v2(row, vector<int>(col));
	k = n - 1;
	i = j = 0;
	v2[i][j] = v1[k--];
	while (k >= 0)
	{
		while (j + 1 != col && !v2[i][j + 1]) v2[i][++j] = v1[k--];
		while (i + 1 != row && !v2[i + 1][j]) v2[++i][j] = v1[k--];
		while (j - 1 != -1 && !v2[i][j - 1]) v2[i][--j] = v1[k--];
		while (i - 1 != -1 && !v2[i - 1][j]) v2[--i][j] = v1[k--];
	}
	for (i = 0; i < row; i++)
	{
		for (j = 0; j < col; j++)
		{
			if (j != 0) printf(" ");
			printf("%d", v2[i][j]);
		}
		printf("\n");
	}
	return 0;
}