PAT乙级 1050 螺旋矩阵

题目描述

原题链接
思路很简单，把它想象为一个人，扛着一些箱子(数)，绕着迷宫走，遇到墙(边界)或者已经有箱子(这个位置不为0，为之前放好的数)就改变方向，而改变方向的顺序是右下左上右…

当然之前先得求出m和n，这个遍历一遍就完事了

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

const int M=10001;
vector<int> a;
int T[M][M]={0};
int n,m;
void check(int& flag,int& i,int& j,int v){
	// cout<<i<<" "<<j<<" "<<v<<" "<<flag<<endl;
	T[i][j]=v;
	if(flag==1){
		j++;if(j>=n||T[i][j]){j--;flag=2;}else{return;}
	}
	if(flag==2){
		i++;if(i>=m||T[i][j]){i--;flag=3;}else{return;}
	}
	if(flag==3){
		j--;if(j<0||T[i][j]){j++;flag=4;}else{return;}
	}
	if(flag==4){
		i--;if(i<0||T[i][j]){i++;flag=1;j++;}else{return;}
	}

}
int main(){
	freopen("input.txt","r",stdin);
	int N;
	cin>>N;
	for(int i=0;i<N;i++){
		int temp;
		cin>>temp;
		a.push_back(temp);
	}
	int r=99999;
	for(int i=N/2+1;i>0;i--){
		if(N%i==0&&(N/i-i)<r&&N/i>=i){
			n=i;m=N/i;
			r=m-n;
		}
	}
	// cout<<m<<" "<<n<<endl;
	sort(a.rbegin(),a.rend());
	// for(auto i:a) cout<<i<<" ";cout<<endl;
	int flag=1;
	int ini_x=0,ini_y=0;
	for(int i=0;i<N;i++){
		check(flag,ini_x,ini_y,a[i]);
	}
	for(int i=0;i<m;i++){
		for(int j=0;j<n-1;j++){
			cout<<T[i][j]<<" ";
		}
		cout<<T[i][n-1]<<endl;
	}
}