https://pintia.cn/problem-sets/994805260223102976/problems/994805275146436608
因为螺旋矩阵有四种方向去填数,所以设置flag=0、1、2、3,对应的有不同的填数方法,同时设置四个方向的边界要不断缩小。
#include<iostream>
#include<algorithm>
#include<math.h>
using namespace std;
int main() {
int a, m, n;
cin >> a;
int num[a];
for(int i=0; i<a; i++){
cin >> num[i];
}
sort(num, num+a);
for(int i=1; i<=sqrt(a); i++){
if(a%i==0){
m = a/i;
n = i;
}
}
int luo[m][n]={0}, i=0, j=0, flag=0;
int row_u=0, row_d=m-1, col_l=0, col_r=n-1;
for(int k=a-1; k>=0; k--){
// 从左向右
if(flag == 0){
luo[i][j++] = num[k];
if(j == col_r){
flag = 1;
row_u++;
}
}
// 从上到下
else if(flag == 1){
luo[i++][j] = num[k];
if(i == row_d){
flag = 2;
col_r--;
}
}
// 从右到左
else if(flag == 2){
luo[i][j--] = num[k];
if(j == col_l){
flag = 3;
row_d--;
}
}
// 从下到上
else if(flag == 3){
luo[i--][j] = num[k];
if(i == row_u){
flag = 0;
col_l++;
}
}
}
for(int i=0; i<m; i++){
for(int j=0; j<n; j++){
cout << luo[i][j];
if(j<n-1){
cout << " ";
}
}
cout << endl;
}
return 0;
}