PAT.A1050 螺旋矩阵

返回目录

样例（可复制）

12
37 76 20 98 76 42 53 95 60 81 58 93

//output
98 95 93
42 37 81
53 20 76
58 60 76

注意点

1. 本题需要解决两个问题：①得到螺旋矩阵的宽col和高row②输出矩阵
2. 对于问题①，由于高度>=宽度，固将高度设置成row=(int)ceil(sqrt(n*1.0));然后不断row++，直到n%row==0。其中ceil向上取整
3. 对于问题②，先将输出的内容打印在二维数组中，然后输出二维数组即可。其中打印二维数组的方式为一圈一圈打印，具体见代码。

#include<bits/stdc++.h>
using namespace std;

const int N=10010;
int arr[N],ans[N][N];
bool cmp(int a,int b){
	return a>b;
}
int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++)scanf("%d",&arr[i]);
	sort(arr,arr+N,cmp);
	int row=(int)ceil(sqrt(n*1.0));
	while(true){
		if(n%row==0)break;
		row++;
	}
	int col=n/row,i=1,j=1,now=0;
	int U=1,D=row,L=1,R=col;
	while(now<n){
        if(now==n-1)ans[i][j]=arr[now++];
		while(now<n&&j<R){//向右 
			ans[i][j++]=arr[now++];
		}
		while(now<n&&i<D){//向下 
			ans[i++][j]=arr[now++];
		}
		while(now<n&&j>L){//向左 
			ans[i][j--]=arr[now++];
		}
		while(now<n&&i>U){//向上 
			ans[i--][j]=arr[now++];
		}
		U++,D--,L++,R--;
		i++,j++;//打印完一圈，进入里面一圈 
	}
	for(int i=1;i<=row;i++){
		for(int j=1;j<=col;j++){
			printf("%d",ans[i][j]);
			if(j!=col)printf(" ");
		}
		printf("\n");
	}
    return 0;
}