B1105 Spiral Matrix (画图)

B1105 Spiral Matrix (25分)

类似题-蛇形方阵

//第一次只拿了21分

• 矩阵的长和宽，求最大因子，从sqrt(num)开始枚举.
• 每次循环一次,s++,t--,d--,r++
• 测试点四运行超时，是因为输入一个数字的时候，需要直接输出这个数字。//1分
• 测试点二运行超时，最后一个数字不必再while循环一次，直接输出即可。//3分

第二个测试点卡了好久/(ㄒoㄒ)/~~

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string.h>
#include <cmath>
#include <math.h>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>
#define lowbit(i)((i)&(-i))
using namespace std;
typedef long long ll;
const int MAX=10010;
const int INF=0x3f3f3f3f;
const int MOD=1000000007;
const int SQR=633;
int mp[1001][1001];
int a[MAX];
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    int num,m,n,k;
    scanf("%d",&num);
    for(int i=1;i<=num;i++)
    {
        scanf("%d",&a[i]);
    }
//测试点4
    if(num==1)
    {
        cout<<a[1]<<endl;
        return 0;
    }
    sort(a+1,a+num+1,cmp);
    k=(int)sqrt(1.0*num);
    while(num%k!=0)
    {
        k++;
    }
    m=k;
    n=num/k;
    if(n>m)
        swap(m,n);
//m行n列
    int d=m,t=n,j=1,now=1,i=1,r=1,s=1;
    while(now<=num)
    {
    while(now<=num&&j<t)
    {
        mp[i][j]=a[now++];
        j++;
    }
    while(now<=num&&i<d)
    {
        mp[i][j]=a[now++];
        i++;
    }
    while(now<=num&&j>r)
    {
        mp[i][j]=a[now++];
        j--;
    }
    while(now<=num&&i>s)
    {
        mp[i][j]=a[now++];
        i--;
    }
    t--,d--,r++,s++;
    j++,i++;
//测试点2
    if(now==num)
        mp[i][j]=a[now++];
    }
    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(j==1)
            cout<<mp[i][j];
            else
            cout<<" "<<mp[i][j];
        }
        cout<<endl;
    }
    return 0;
}