PAT Basic1050 螺旋矩阵

版权声明：个人学习笔记记录 https://blog.csdn.net/Ratina/article/details/84657069

链接：PAT Basic1050

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

---

该题方法并不难想到，我这里采用的方法是设定 上下左右边界（u,d,l,r），每绕完一圈，u++，d–，l++，r–，以此形成螺旋。
但是因为没注意一些细节，WA了很多次…( WA到怀疑人生… )
首先是存储矩阵的二维数组，因为没给出N的范围，所以这里要动态定义二维数组。
用vector定义动态 一维数组：

vector< int > ar ( n,value );　　　　　//对应 ar[n]，value为初值(可省略)

用vector定义动态 二维数组：

vector< vector< int > > ar (m , vector< int > (n ,value ) ); 　 // 对应ar[m][n]

---

还有些要注意的点，看代码:

#include<algorithm>
#include<cstdio>
#include<cmath>
#include<vector>
using namespace std;
int main()
{
	int N, ar[10010];
	int i, j, m, n;
	scanf("%d", &N);
	for (i = 0; i < N; i++)   //输入、排序
		scanf("%d", &ar[i]);
	sort(ar, ar + N);
	for (i = 1; i <= sqrt(N); i++)  //按题目要求找到 m、n
	{
		if (N%i == 0)
		{
			n = i;
			m = N / i;
		}
	}
	vector<vector<int> > output(m + 1, vector<int>(n + 1));  //动态二维数组
	int u = 1, d = m, l = 1, r = n, k = N - 1;
	while (k != -1)
	{
		if (u==d&&l==r&&k!=-1)    //注意！当u==d&&l==r，下面4个for均不会执行
			output[u][l] = ar[k--];
		for (i = l; i < r&&k!=-1; i++)  //for的条件要有 k!=-1,不然元素用完了还在继续填充！
			output[u][i] = ar[k--];
		for (i = u; i < d&&k!=-1; i++)
			output[i][r] = ar[k--];
		for (i = r; i > l&&k!=-1; i--)
			output[d][i] = ar[k--];
		for (i = d; i > u&&k!=-1; i--)
			output[i][l] = ar[k--];
		u++;
		d--;
		l++;
		r--;
	}
	for (i = 1; i <= m; i++)
	{
		for (j = 1; j <= n; j++)
		{
			if (j != 1)
				printf(" ");
			printf("%d", output[i][j]);
		}
		printf("\n");
	}
	return 0;
}