1050 螺旋矩阵(25 分)
本题要求将给定的 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的矩阵来说:
第一部分,位于第i行,坐标从[i][i]开始,到[i][n - i - 1]为止。
第二部分,位于第n - i - 1列,坐标从[i + 1][n - i - 1]开始,到[m - i - 2][n - i - 1]为止。
第三部分,位于第m - i - 1行,坐标从[m - i - 1][n - i - 1]开始,到[m - i - 1][i]为止。
第四部分,位于第i列,坐标从[m - i - 2][i]开始,到[i + 1][i]为止。
代码:
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
bool cmp(int a, int b) {
return a > b;
}
int main() {
int N;
cin >> N;
int m, n;
int temp = sqrt((double)N);
while (temp != 0) {
if (N % temp == 0) {
n = temp;
break;
}
temp--;
}
m = N / n;
vector<int> v(N);
for (int i = 0; i < N; i++) {
cin >> v[i];
}
sort(v.begin(), v.end(), cmp);
int level = m / 2 + m % 2;
int index = 0;
vector<vector<int> > arr(m, vector<int>(n));
for (int i = 0; i < level; i++) {
for (int j = i; j <= n - i - 1 && index < N; j++) {
arr[i][j] = v[index++];
}
for (int j = i + 1; j <= m - i - 2 && index < N; j++) {
arr[j][n - i - 1] = v[index++];
}
for (int j = n - i - 1; j >= i && index < N; j--) {
arr[m - 1 - i][j] = v[index++];
}
for (int j = m - i - 2; j >= i + 1 && index < N; j--) {
arr[j][i] = v[index++];
}
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (j != n - 1) {
cout << arr[i][j] << ' ';
} else {
cout << arr[i][j] << endl;
}
}
}
return 0;
}