C++ - PAT - 1050. 螺旋矩阵(25)

本题要求将给定的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

思路：

1.求出n和m的值

2.用二维数组来存储答案，最后数序输出。

3.重点是如何对二位数组进行赋值，从而达到螺旋非增序列。

4.方法就是按照，右，下，左，上的顺序，进行判断是否已经赋值和是否越界，然后进行数值的更新。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;

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

int N ;
int main(){
	scanf("%d",&N);
	int an,am;
	int m,n;
	int min = 1000000;
	
	int num[N];
	for(int i=0 ;i<N ;i++){
		scanf("%d",&num[i]);
	} 
	
	//遍历，求出n和m 
	for(int i=1 ;i<=sqrt(N) ;i++){
		int temp = N;
		if(N%i==0){
			n = i;
			m = temp/i;
		 	if(m>=n&&m-n<min){
		 		min = m-n;
				an = n;
				am = m;	
			}else{
				continue;
			}
		}
	}
	//降序排列 
	sort(num,num+N,cmp);
	
	int ans[am][an];//用来存储要输出的数据 
	
	//赋初值-1 
	for(int i=0 ;i<am ;i++){
		for(int j=0 ;j<an ;j++){
			ans[i][j]=-1;
		}
	}
	
	
	int a1=0,a2=0;

	bool right=true,down=false,left=false,up=false;
	for(int i=0 ;i<N ;i++){
		//1 右 
		if(right){
			if(a2<an&&ans[a1][a2]==-1){
				ans[a1][a2] = num[i];
				a2++;
				continue;
			}else{
				down = true;
				right = false;
				a2--;
				a1++;
			}
		}
		
		//2	下 
		if(down){
			if(a1<am&&ans[a1][a2]==-1){
				ans[a1][a2] = num[i];
				a1++;
				continue;
			}else{
				down = false;
				left = true;
				a1--;
				a2--;	
			}	
		}
		
		//3	左 
		if(left){
			if(a2>=0&&ans[a1][a2]==-1){
				ans[a1][a2] = num[i];
				a2--;
				continue;
			}else{
				left = false;
				up = true;
				a2++;
				a1--;
			}
		}
		
		//4	上 
		if(up){
			if(a1>=0&&ans[a1][a2]==-1){
				ans[a1][a2] = num[i];
				a1--;
				continue;
			}else{
				up = false;
				right = true;
				a1++;
				a2++;		
			}	
		}
		//再跟一个右，如果第一次up==false right==true更新后
		//因为写的是顺序语句，所以不会返回去进行判断，直接进行下一次循环，导致结果出现错误。 
		//1 右 
		if(right){
			if(a2<an&&ans[a1][a2]==-1){
				ans[a1][a2] = num[i];
				a2++;
				continue;
			}else{
				down = true;
				right = false;
				a2--;
				a1++;
			}
		}
	
	}
	
	
	for(int i=0 ;i<am ;i++){
		for(int j=0 ;j<an ;j++){
			if(j==0){
				printf("%d",ans[i][j]);
			}else{
				printf(" %d",ans[i][j]);
			}
		} 
		printf("\n");
	}
	
	return 0;
}