PAT乙级—1050

#include <iostream>
#include <vector>
#include <algorithm>
#include <math.h>
using namespace std;
bool comp(int a, int b){
return a>b;
}
int main() {
int N;
int m=0, n=0;
cin>>N;
for(int i=(int)sqrt(N); i<=N; i++){
if(N%i==0){
m=i;
n=N/m;
break;
}
}
cout<<n<<' '<<m<<endl;
vector<int>v;
int a;
for(int i=0; i<N; i++){
cin>>a;
v.push_back(a);
}
sort(v.begin(), v.end(), comp);
int x=0, y=0, j=0;
vector<vector <int>>b;
//int b[1000][1000]={0};
while(j<N){
while(y+1<n&&!b[x][y+1]) b[x][++y]=v[++j];
while(x+1<m&&!b[x+1][y]) b[++x][y]=v[++j];
while(y-1>=0&&!b[x][y-1]) b[x][--y]=v[++j];
while(x-1>=0&&!b[x-1][y]) b[--x][y]=v[++j];
}
for(int i=0; i<m; i++){
for(int h=0; h<n; h++){
if(h==n-1) cout<<b[i][h]<<endl;
else cout<<b[i][h]<<' ';
}
}
return 0;
}