PAT (Basic Level) Practice （中文） B1050 螺旋矩阵 （25 分）（C++）（甲级）

原题链接：B1050 螺旋矩阵

#include<algorithm>
#include<iostream>
#include<cstdio>
#include <cmath>
using namespace std;

#define lowbit(x) ((x)&(-(x)))

const int MAX = 10010, SQRT = 105;
int N, A[MAX], m, n;
int T[MAX][SQRT] = {{0}};//注意此时T的大小，列数只要100多点就行了，太大会导致内存不够用

bool cmp(int a, int b)
{
    return a > b;
}

int main()
{
    scanf("%d", &N);
    for(int i=0; i<N; i++) scanf("%d", &A[i]);
    sort(A, A+N, cmp);//从大到小排序
    for(m=(int)ceil(sqrt(N)); m<=N && N%m; m++);//求m/n
    n = N/m;
    int cnt = 0;
    int U = 1, D = m, L = 1, R = n;
    while(cnt < N)
    {
        if(L == R && U == D) T[U][L] = A[cnt++];//最后汇聚到一点时需要单独考虑，通过下面四条无法赋值该点
        int i = L;
        while(cnt < N && i < R) T[U][i++] = A[cnt++];//上面那排
        i = U;
        while(cnt < N && i < D) T[i++][R] = A[cnt++];//最后一列
        i = R;
        while(cnt < N && i > L) T[D][i--] = A[cnt++];//最底层
        i = D;
        while(cnt < N && i > U) T[i--][L] = A[cnt++];//最左侧
        U++, D--, L++, R--;
    }
    for(int i=1; i<=m; i++)//打印
    {
        printf("%d", T[i][1]);
        for(int j=2; j<=n; j++) printf(" %d", T[i][j]);
        printf("\n");
    }
    return 0;
}