【PAT】(B)1050 螺旋矩阵(25)

『题目』


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

输入格式:

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

输出格式:

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

『思路』


本题的思路并不难,是一个简单的模拟。 先输入N,找出N的两个差值最小的约数,就是sqrt(N)前后的两个数,大的为m,小的为n.然后将输入的数组降序排列,最后也是本题的核心,蛇形填数,也就是题目中的 螺旋矩阵,只需要按着向右 向下 向左 向上这四个方向 一直循环填数就可以,同时开始将数组初始化为0,表示没有填数,通过判断下一步是否已经填数来决定是否跳出循环.

『代码』(第一次)


#include <iostream>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
int m,n,N;
bool cmp(const int& a,const int& b){
	return a>b;
}
void getmn(int x) {
	int t;
	for(int i = 1; i <= sqrt(N); ++i)
		if(N%i == 0)
			t = i;
	n = t;
	m = N / t;
}
int main() {
	cin>>N;
	int a[N+1],flag = 0,bk = 1;
	for(int i = 0; i < N; ++i)  //9 1 2 3 4 5 6 7 8 9
		cin>>a[i];
	sort(a,a+N,cmp);
	getmn(N);
	int book[100][100];
	memset(book,0,sizeof(book));
	int i = 1,j = 1;
	while(flag < N) {//一直循环 向右 向下 向左 向上这四个方向 
		//一直向右走
		for(; j < n; j++) {
			if(book[i][j] == 0)
				book[i][j] =  a[flag++];
			if(book[i][j+1] != 0)//判断下一步是否越界
				break;
		}
		//一直向下走
		for(; i < m; i++) {
			if(book[i][j] == 0)
				book[i][j] = a[flag++];
			if(book[i+1][j] != 0)
				break;
		}
		//一直向左走
		for(; j > 1; j--) {
			if(book[i][j] == 0)
				book[i][j] = a[flag++];
			if(book[i][j-1] != 0)
				break;
		}
		//一直向上走
		for(; i > 1; i--) {
			if(book[i][j] == 0)
				book[i][j] = a[flag++];
			if(book[i-1][j] != 0) 
				break;
		}
	}
	for(i = 1; i <= m; i++){
		for(j = 1; j <= n; j++){
			printf("%02d",book[i][j]);
			if(j != 3)
				printf(" ");
		}
		printf("\n");	
	}
	return 0;
}

第一次的代码 OJ显示多种错误,数据格式错误,超时, orz orz ....

后来把输出格式改了一下,就只剩下超时的问题,优化代码 ,翻出大紫书,到第四十页,嘿嘿嘿....

发现之前的方法并不好,多次判断,必然会超时,并且在求m和n的地方存在一些问题,大紫书 上是判断下一个位置并直接填下一个位置里的数,妙

『AC代码』 

#include<bits/stdc++.h>
using namespace std; 
int a[10000][1000]={0},s[10000]; 
bool cmp(int a,int b){
    return a>b;
}
int main(){
    int n,i,j,x,y,r,c,tot,minn=9999;
   
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    scanf("%d",&s[i]);
    sort(s,s+n,cmp);
    for(i=1;i<=sqrt(n*1.0);i++)
    {
        if(n%i==0)
        {
            if(n/i-i<minn){
                minn=n/i-i;
                r=i;
            }
        }        
    }
    c=n/r;//c>r c行r列 
    a[1][1]=s[0];
    tot=0;
	x=y=1;
    while(tot < r * c-1)
    {
        while(y + 1 <= r && ! a[x][y + 1])
            a[x][++y] = s[++tot];
        while(x + 1 <= c && !a[x + 1][y])
            a[++x][y] = s[++tot];
        while(y - 1 > 0 && !a[x][y - 1])
            a[x][--y] = s[++tot];
        while(x - 1 > 0 && !a[x - 1][y])
            a[--x][y] = s[++tot];   
    }
    for(i=1;i<=c;i++){
            printf("%d",a[i][1]);
        for(j=2;j<=r;j++){
                printf(" %d",a[i][j]);
        } 
        printf("\n");
        
    }
    return 0;
}

『感悟』

大紫书上的代码是真滴厉害

猜你喜欢

转载自blog.csdn.net/sinat_40872274/article/details/81275003