B1050. 螺旋矩阵(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

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

int N;
int pos = 0;
int ans[10000][10000] = { 0 };

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

int main()
{
    vector<int> input;
    int temp;
    int row, col;
    int flag = 0;
    int  you = 0;//本来想着上下左右的，结果只用一个“右”就搞定

    cin >> N;
    for (int i = 0; i < N; ++i)
    {
        cin >> temp;
        input.push_back(temp);
    }
    sort(input.begin(), input.end(),cmp);
    //计算行列
    for (int i = sqrt(N); i >=1; --i)
    {
        if (N%i == 0)
        {
            col = i;
            row = N / i;
            break;
        }
    }
    //相应位置放入数据
    while (pos < N)
    {
        for (int i = you; i < col - you&&flag==0; ++i)
        {
            ans[you][i] = input[pos];
            ++pos;
            if (pos==N)
            {
                flag = 1;
                break;
            }
        }
        ++you;
        for (int i = you; i < row - you&&flag == 0; ++i)
        {
            ans[i][col - you] = input[pos];
            ++pos;
            if (pos == N)
            {
                flag = 1;
                break;
            }
        }
        for (int i = col - you; i >= you-1 && flag == 0; --i)
        {

            ans[row-you][i] = input[pos];
            ++pos;
            if (pos == N)
            {
                flag = 1;
                break;
            }
        }

        for (int i = row - you - 1 ; i >= you&& flag == 0; --i)
        {
            ans[i][you - 1] = input[pos];
            ++pos;
            if (pos == N)
            {
                flag = 1;
                break;
            }
        }
    }
    //输出
    for (int i = 0; i < row; ++i)
    {
        for (int j = 0; j < col; ++j)
        {
            if (j == 0)
                cout << ans[i][j];
            else
                cout << ' ' << ans[i][j];
        }
        cout << endl;
    }


    system("pause");
    return 0;
}

为了方便理解，说下数据填入的模式：
1 1 1 1 1 1 1
4 5 5 5 5 5 2
4 8 9 9 9 6 2
4 8 a a a 6 2
4 7 7 7 7 7 2
3 3 3 3 3 3 3
就像上下夹层，把中间夹住