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

1050. 螺旋矩阵(25)

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行n个数字，共m行。相邻数字以1个空格分隔，行末不得有多余空格。

输入样例：1237 76 20 98 76 42 53 95 60 81 58 93输出样例：98 95 9342 37 8153 20 7658 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;}