[PAT乙级]1050. 螺旋矩阵(25)

1050. 螺旋矩阵(25)

原题链接
本题要求将给定的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

注意：

• while里面的条件要注意，很重要，之前条件没写对，循环一直进不去。。。很伤

代码：

#include <iostream>
#include <cmath>
#include <vector>
#include <algorithm>
using namespace std;
int cmp(int a, int b){
    return a > b;
}
int findn(int N){
    int i = sqrt((double)N);//i是小于N的平方根并最接近N的平方根整数
    while(i >= 1){
        if(N%i == 0)
            return i;
        i--;
    }
    return 1;
}
int main()
{
    int N;
    cin >> N;
    int n = findn(N);
    int m = N/n;
    vector<int> arr(N);
    for(int i=0; i<N; i++)
        cin >> arr[i];
    sort(arr.begin(), arr.end(), cmp);//降序排列
    vector<vector<int>> res(m, vector<int>(n));
    for(int i=0; i<m; i++){
        for(int j=0; j<n; j++){
            res[i][j] = 0;
        }
    }
    int i=0,j=0,k=0;
    res[i][j] = arr[k];//为了以后赋值方便先将第一个元素赋值
    k++;
    while(k < N){
        while(j<n-1 && res[i][j+1]==0){//往右赋值 && 即将赋值的位置没被赋过值
            j++;//放在这个位置，省去下一个while循环的j--
            res[i][j] = arr[k];
            k++;
        }
        while(i<m-1 && res[i+1][j]==0){//往下赋值 && 即将赋值的位置没被赋过值
            i++;
            res[i][j] = arr[k];
            k++;
        }
        while(j>0 && res[i][j-1]==0){//往左赋值 && 即将赋值的位置没被赋过值
            j--;
            res[i][j] = arr[k];
            k++;
        }
        while(i>0 && res[i-1][j]==0){//往上赋值 && 即将赋值的位置没被赋过值
            i--;
            res[i][j] = arr[k];
            k++;
        }
    }
    for(int i=0; i<m; i++){
        for(int j=0; j<n; j++){
            if(j != 0)
                cout <<" ";
            cout << res[i][j];
        }
        cout << endl;
    }
    return 0;
}