PAT | A1105 Spiral Matrix（待优化）

测试点1和3答案错误，看不出来哪错了。。

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>

using namespace std;

vector<int> seq;

int res[1000][1000];

bool cmp(int a,int b){
	return a > b;
}

void fillRes(int i,int m,int n,int k){ // 从数组的第i个元素开始填，此时递归到第k层
	// 从(k,k)到(m,n)的矩形
	if(m < k || n < k)
		return;
	for(int j = k;j <= n;j++)
		res[k][j] = seq[i++];
	for(int j = k + 1;j <= m;j++)
		res[j][n] = seq[i++];
	for(int j = n - 1;j >= k;j--)
		res[m][j] = seq[i++];
	for(int j = m - 1;j > k;j--)
		res[j][k] = seq[i++];
	fillRes(i,m - 1,n - 1,k + 1);
	return;
}

int main(){
	int N;
	scanf("%d",&N);
	for(int i = 0;i < N;i++){
		int temp;
		scanf("%d",&temp);
		seq.push_back(temp);
	}
	if(N == 0){
		printf("0");
		system("pause");
		return 0;
	}
	sort(seq.begin(),seq.end(),cmp);
	int x = sqrt(N);
	while(N % x != 0){
		x++;
	}
	int m = x,n = N / m;
	if(m < n)
		swap(m,n);
	fillRes(0,m - 1,n - 1,0);
	for(int i = 0;i < m;i++){
		for(int j = 0;j < n;j++){
			printf("%d",res[i][j]);
			if(j != n - 1)
				printf(" ");
		}
		if(i != m - 1)
			printf("\n");
	}
	system("pause");
	return 0;
}