PAT (Basic Level) Practice 1050 螺旋矩阵 (25分) (数组标记、内存使用)
1.题目
本题要求将给定的 N 个正整数按非递增的顺序,填入“螺旋矩阵”。所谓“螺旋矩阵”,是指从左上角第 1 个格子开始,按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列,满足条件:m×n 等于 N;m≥n;且 m−n 取所有可能值中的最小值。
输入格式:
输入在第 1 行中给出一个正整数 N,第 2 行给出 N 个待填充的正整数。所有数字不超过 104,相邻数字以空格分隔。
输出格式:
输出螺旋矩阵。每行 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
2.题目分析
首先按照求一个数的素数的原理,从N的开方开始向前找,第一个可以被整除的数就是n,num/n就是m
将数字排序后就可以一圈一圈从外向内放数字,遇到边界跳过,这样从外向内遍历,for循环中的界限不用变化,在内圈输出遇到前面外圈已经输出的数字,直接跳过就好
开始使用了两个二维数组,一个输出,一个标记,结果最后一个测试点内存超限,改进使用一个二维数组直接输出
3.代码
#include<iostream>
#include<algorithm>
#include<cmath>
#include<functional>
using namespace std;
int out[10001][10001],/*mark[10001][10001],*/list[10001];
int main()
{
int num;
cin >> num;
for (int i = 0; i < num; i++)
cin >> list[i];
int m, n;
for (int i = sqrt(num); i >= 1;i--)
if (num%i == 0) { n = i; m = num / n; break; }
sort(list, list + num,greater<int>());
int round = 0;
int count = 0;
while(count!=num)
{
for (int j = 0; j < n; j++)
{
if (out[round][j] != 0)continue;
//if(mark[round][j] == 1)continue;
out[round][j] = list[count++];
//mark[round][j] = 1;
}
for (int j = 0; j < m; j++)
{
if (out[j][n - 1 - round] != 0)continue;
//if (mark[j][n - 1 - round] ==1)continue;
out[j][n - 1 - round] = list[count++];
//mark[j][n - 1 - round] =1;
}
for (int j = n - 1; j >= 0; j--)
{
if (out[m - 1 - round][j] != 0)continue;
//if (mark[m - 1 - round][j] ==1)continue;
out[m - 1 - round][j] = list[count++];
//mark[m - 1 - round][j] = 1;
}
for (int j = m - 1; j >= 0; j--)
{
if (out[j][round] != 0)continue;
//if (mark[j][round] ==1)continue;
out[j][round] = list[count++];
//mark[j][round] = 1;
}
round++;
}
int space = 0;
for (int i = 0; i < m; i++)
{
space = 0;
for (int j = 0; j < n; j++)
{
if (space == 0) { cout << out[i][j]; space++; }
else cout << ' ' << out[i][j];
}
cout << endl;
}
}