PAT 乙 1050 螺旋矩阵（1段错误）

1050 螺旋矩阵 （25 分）

本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列，满足条件：m×n 等于 N；m≥n；且 m−n 取所有可能值中的最小值。

输入格式：

输入在第 1 行中给出一个正整数 N，第 2 行给出 N 个待填充的正整数。所有数字不超过 10​4​​，相邻数字以空格分隔。

输出格式：

输出螺旋矩阵。每行 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用sqrt（）会出现多个超时，不得已借鉴了一下

#include<iostream>
#include<math.h>
#include<algorithm>
using namespace std;
bool cmp(int a,int b)
{return a>b;
}
int func(int N) {
    int i = sqrt((double)N);
    while(i >= 1) {
        if(N % i == 0)
            return i;
        i--;
    }
    return 1;
}
int main()
{ int n;
static int data[100000];
  scanf("%d",&n);
  for(int i=0;i<n;i++)
  { scanf("%d",&data[i]);}
sort(data,&data[n],cmp);
//  for(int i=0;i<n;i++)
//  {cout<<data[i]<<endl;}
int indux[500][500]={0};
int cnt=0,hang=1,lei=0;
int M;M=func(n);
int N=n/M;
while(cnt<n)
{
	while(indux[hang][lei+1]==0&&lei<=M-1){
		indux[hang][lei+1]=data[cnt];
		lei=lei+1;cnt++;
//		cout<<"hang:"<<hang<<" "<<"lei:"<<lei<<"cout:"<<cnt<<endl;
	}//右填充

	while(indux[hang+1][lei]==0&&hang<=N-1){
	indux[hang+1][lei]=data[cnt];
	hang=hang+1;cnt++;
//	cout<<"hang:"<<hang<<" "<<"lei:"<<lei<<"cout:"<<cnt<<endl;
	}//下填充

	while(indux[hang][lei-1]==0&&lei>1){
	indux[hang][lei-1]=data[cnt];
	lei=lei-1;cnt++;
//	cout<<"hang:"<<hang<<" "<<"lei:"<<lei<<"cout:"<<cnt<<endl;
	}//左填充

	while(indux[hang-1][lei]==0&&hang>1){
	indux[hang-1][lei]=data[cnt];
	hang=hang-1;cnt++;
	}//上填充
if(cnt==n)break;
}
for(int i=1;i<=N;i++)
{ for(int j=1;j<=M;j++)
 if(j==1)printf("%d",indux[i][j]);
 else
 printf(" %d",indux[i][j]);
if(i!=N)
printf("\n");
}


}