【算法笔记】B1050 螺旋矩阵

1050 螺旋矩阵 （25 分)

 

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

输入格式：

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

输出格式：

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

 

【碰壁法】判断下一个数，如果不为0就拐弯。如图：

   

　　最后一个测试点总是超时，把 col = (int)sqrt(n) 改成  col = sqrt(n) 后 AC。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 10010;
int a[maxn][maxn] = {0};
bool cmp(int a, int b){
    return a > b;
}
int main(){
    int n, ori[maxn];
    cin>>n;
    int col = sqrt(n), row;
    while(n % col){
        col--;
    }
    row = n / col;
    for(int i = 0; i < n; i++){
        scanf("%d",&ori[i]);
    }
    sort(ori, ori + n, cmp);
    int x = 1, y = 1, cnt = 0;
    a[y][x] = ori[cnt];
    while(cnt + 1 < n){
        while(y + 1 <= col && ! a[x][y + 1])
            a[x][++y] = ori[++cnt];
        while(x + 1 <= row && !a[x + 1][y])
            a[++x][y] = ori[++cnt];
        while(y - 1 > 0 && !a[x][y - 1])
            a[x][--y] = ori[++cnt];
        while(x - 1 > 0 && !a[x - 1][y])
            a[--x][y] = ori[++cnt];  
    }
    for(int i=1;i<=row;i++){
        printf("%d",a[i][1]);
        for(int j=2;j<=col;j++){
            printf(" %d",a[i][j]);
        } 
        printf("\n");
    }
    return 0;
}