题目描述
本题要求将给定的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
题目解析
其实最好是把各个功能模块单独写成子函数,这样调试比较方便;开始写主函数的框架的时候可以把子函数先给一个正确的返回值,这样就可以先测试其他部分是否正确,最后再把这个功能子函数实现。
- 先用一个数组读入给出的数列,进行排序;
- 找出合适的m和n;
- 再建立一个数组,把排序好的数列按照螺旋填进去;
- 输出矩阵。
发现一个更好的填螺旋的方法,见@luoluo
代码
#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;
int main()
{
int N, m, n, val;
vector<int> list;
cin >> N;
for (int i = 0; i < N; ++i)
{
cin >> val;
list.push_back(val);
}
sort(list.begin(), list.end(), [](int a, int b) {return a > b; });
m = n = sqrt(N);
while (m*n != N)
{
if (m*n < N)
{
++m;
}
else
{
--n;
}
}
vector<int> out(N,0);
string direct = "n+";
for (int i = 0,a=0,b=0,m0=0,n0=0,m1=m,n1=n; i < N; )
{
if (direct=="n+")
{
out[a*n + b] = list[i];
++b;
++i;
if (b == n1)
{
direct = "m+";
++m0;
++a;
--b;
}
continue;
}
if (direct == "m+")
{
out[a*n + b] = list[i];
++a;
++i;
if (a == m1)
{
direct = "n-";
--n1;
--b;
--a;
}
continue;
}
if (direct == "n-")
{
out[a*n + b] = list[i];
--b;
++i;
if (b == n0 - 1)
{
direct = "m-";
--m1;
--a;
++b;
}
continue;
}
if (direct == "m-")
{
out[a*n + b] = list[i];
--a;
++i;
if (a == m0 - 1)
{
direct = "n+";
++n0;
++b;
++a;
}
continue;
}
}
for (int i = 1; i <= N; ++i)
{
cout << out[i-1];
if (i%n == 0)
{
cout << endl;
}
else
{
cout << " ";
}
}
system("pause");
return 0;
}