PAT BASIC LEVEL 1050. 螺旋矩阵(25)

1050. 螺旋矩阵(25)

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

输入格式：

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

输出格式：

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

Answer:

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
void print(int *nums, int *mat, int m, int n) {
    for(int i=0; i<m; i++) {
        for(int j=0; j<n; j++) {
            if(j)
                cout << ' ' << nums[mat[i*n+j]];
            else
                cout << nums[mat[i*n+j]];
        }
        cout << '\n';
    }
}

int main() {
    int tol, m, n;
    cin >> tol;
    int nums[tol];
    for(int i=0; i<tol; i++) {
        cin >> nums[i];
    }
    double t = sqrt(tol);
    m = t > int(t) ? t+1 : t;
    while(tol%m) m++;
    n = tol/m;
    sort(nums, nums+tol);
    int mat[m*n] = {};
    int max = m*n-1, dif = 0;
    int up, down;
    for(int i=0; i<m/2; i++) {
        up = max - dif;
        dif += (m+n-2)*2-(i*8);
        down = max-dif + (m - 2*i - 1);
        for(int j=i; j<n-i; j++) {
            mat[i*n+j] = up--;
            mat[(m-1-i)*n+j]=down++;
        }
    }
    dif=0;
    int left, right;
    int mid = n%2 ? n/2+1 : n/2;
    for(int i=0; i<mid; i++) {
        right = max - (dif+(n-i*2));
        dif += (m+n-2)*2-(i*8);
        left = max-dif;
        for(int j=i+1; j<m-i-1; j++) {
            mat[j*n+i] = ++left;
            mat[n*(j+1)-i-1]=right--;
        }
    }
    print(nums, mat, m, n);
}

PS.
毕竟线代什么的。
还是补课去吧→_→