PAT 1050. 螺旋矩阵

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

输入格式:

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

输出格式:

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

阅读更多
个人分类: PAT
上一篇PAT 1049. 数列的片段和
下一篇PAT 1051. 复数乘法
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭