PAT 1050. 螺旋矩阵

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行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

这道题在程序竞赛上碰到过，比这个简单可是写的很混乱，这次又卡在了这题上很久，不过不是在逻辑上，而是总是有三个测试

点超时，看了好多别人的代码，把自己的代码改的面目全非结果还是超时。最后问题居然出在行数和列数的求解上，真是莫名其

妙。

#include <iostream>
#include <cmath>
#include <algorithm>
#include <vector>
using namespace std;
//直接令n=(int)sqrt(N)就总是有三个测试点超时，改用这个函数 就ok了，神奇，，， 
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;
	scanf("%d",&N);
	int m,n;
	n=func(N);
	m=N/n;
	vector<int>num(N);
	for(int i=0;i<N;i++){
		scanf("%d",&num[i]);
	}
	sort(num.begin(),num.end());
	vector<vector<int> > mart(m);
	for(int i = 0 ; i < m ; i++){
		mart[i].assign(n,0);
	} 
	int p=0,r=-1,k=N-1;
	while(k>=0){
		while(r+1<n&&!mart[p][r+1]){
			mart[p][++r]=num[k--];
		}

	    while(p+1<m&&!mart[p+1][r]){
			mart[++p][r]=num[k--];
		}

		while(r-1>=0&&!mart[p][r-1]){
			mart[p][--r]=num[k--];
		}

		while(p-1>=0&&!mart[p-1][r]){
			mart[--p][r]=num[k--];
		}
	}
	int jj;
	for(int i=0;i<m;i++){
		for(jj=0;jj<n-1;jj++){
			printf("%d ",mart[i][jj]);
		}
		printf("%d\n",mart[i][jj]);
	}
	return 0;
}