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
注意:
- while里面的条件要注意,很重要,之前条件没写对,循环一直进不去。。。很伤
代码:
#include <iostream>
#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
int cmp(int a, int b){
return a > b;
}
int findn(int N){
int i = sqrt((double)N);//i是小于N的平方根并最接近N的平方根整数
while(i >= 1){
if(N%i == 0)
return i;
i--;
}
return 1;
}
int main()
{
int N;
cin >> N;
int n = findn(N);
int m = N/n;
vector<int> arr(N);
for(int i=0; i<N; i++)
cin >> arr[i];
sort(arr.begin(), arr.end(), cmp);//降序排列
vector<vector<int>> res(m, vector<int>(n));
for(int i=0; i<m; i++){
for(int j=0; j<n; j++){
res[i][j] = 0;
}
}
int i=0,j=0,k=0;
res[i][j] = arr[k];//为了以后赋值方便先将第一个元素赋值
k++;
while(k < N){
while(j<n-1 && res[i][j+1]==0){//往右赋值 && 即将赋值的位置没被赋过值
j++;//放在这个位置,省去下一个while循环的j--
res[i][j] = arr[k];
k++;
}
while(i<m-1 && res[i+1][j]==0){//往下赋值 && 即将赋值的位置没被赋过值
i++;
res[i][j] = arr[k];
k++;
}
while(j>0 && res[i][j-1]==0){//往左赋值 && 即将赋值的位置没被赋过值
j--;
res[i][j] = arr[k];
k++;
}
while(i>0 && res[i-1][j]==0){//往上赋值 && 即将赋值的位置没被赋过值
i--;
res[i][j] = arr[k];
k++;
}
}
for(int i=0; i<m; i++){
for(int j=0; j<n; j++){
if(j != 0)
cout <<" ";
cout << res[i][j];
}
cout << endl;
}
return 0;
}