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

IDEA

1.首先对序列从大到小排序；

2.求出差值最小的row和col；

3.存储螺旋矩阵，声明一个二维矩阵  ，动态分配空间，递归存储（上右下左顺序）

CODE

#include<iostream>
#include<vector> 
#include<algorithm>
using namespace std;
int cmp(int a,int b){
  return a>=b;
}
int k=0;
void setMatrix(int **matrix,int x,int y,int row,int col,vector<int> vec){
  int i,j;
  if(row<=0||col<=0){//判断row，col减到0，则退出系统 
    return;
  }
  if(col==1){// 判断，如果col是1，则输出一列 
    for(i=y;i<y+row;i++){
      matrix[i][x]=vec[k++];   
    }
    return;
  }
  for(i=x;i<x+col-1;i++)//上边   
    matrix[y][i]=vec[k++]; 
  for(j=y;j<y+row-1;j++)//右边 
    matrix[j][x+col-1]=vec[k++];
  for(i=x+col-1;i>x;i--)//下边 
    matrix[y+row-1][i]=vec[k++];
  for(j=y+row-1;j>y;j--)//左边 
    matrix[j][x]=vec[k++];
  setMatrix(matrix,x+1,y+1,row-2,col-2,vec);//递归 
}
int main(){
  int n;
  cin>>n;
  vector<int> vec;
  for(int k=0;k<n;k++){
    int a;
    cin>>a;
    vec.push_back(a);
  }
  sort(vec.begin(),vec.end(),cmp);
  int row=n,col=1,min=n-1;
  //求行数row 列数col 
  for(int i=n,j=1;i>=j;i--){
    if(n%i==0){
      j=n/i;
      if(i>=j&&min>i-j){
        min=i-j;
        row=i;
        col=j;
      }
    }
  }
  //cout<<"row="<<row<<" "<<"col="<<col<<endl;
  int **matrix;
  matrix=(int **)malloc(row*sizeof(int *));
  for(int i=0;i<row;i++)
    matrix[i]=(int *)malloc(col*sizeof(int));
    
  int start=1;
  setMatrix(matrix,0,0,row,col,vec);
  for(int i=0;i<row;i++){
    for(int j=0;j<col;j++){
      if(j==0){
        cout<<matrix[i][j];
      }else{
        cout<<" "<<matrix[i][j];
      }    
    }
    cout<<endl;
  }
  return 0;
}