1050 螺旋矩阵 (25 分)--PAT乙级

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

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

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

法一：dfs

分析：首先，根据N求n,m。从根号N~1遍历寻找n*m = N,跳出。通过dfs逐个写入矩阵中。这种方法比较容易想到，不过会爆栈QAQ。第7个测试点不能过。

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
int N, n, m;
int num[1000010];
int square[10010][10010];
bool book[10010][10010];
//顺时针：右下左上
int k = 0;
bool flag = false;
int nxt[4][2] = {{0,1},{1,0},{0,-1},{-1,0}};

void dfs(int x, int y, int step){
    if(step >= N || flag == true){
    	flag = true;
        return;
    }
    int tx = x + nxt[k % 4][0], ty = y + nxt[k % 4][1];
    if(tx < 0 || tx >= n || ty < 0 || ty >= m || book[tx][ty] != 0){
    	k ++;
    	dfs(x, y, step);
    	return;
	}
	square[tx][ty] = num[step];
	book[tx][ty] = 1;
	dfs(tx, ty, step + 1);	
}

bool cmp(int x, int y){
    return x > y;
}

int main(){
    cin >> N;
    for(int i = 0; i < N; i ++){
        cin >> num[i];
    }
    sort(num, num + N, cmp);
    for(int i = sqrt(N); i > 0; i --){
        if((float)N / (float)i == N / i){
            m = i;
            n = N / i;
            break;
        }
    }
    square[0][0] = num[0];
    book[0][0] = 1;
    dfs(0, 0, 1);
    
    for(int i = 0; i < n; i ++){
        for(int j = 0; j < m - 1; j ++){
            cout << square[i][j] << ' ';
        }
        cout << square[i][m - 1] << endl;
    }
    return 0;
}

法二：循环

为了解决爆栈，不能使用dfs，只能采取遍历的方式。

1. 首先求外层循环需要多少次，m>=n，所以肯定与m有关，为m/2+m%2次。
2. 接着分析每次循环的内容
   

#include<iostream>
#include<algorithm>
#include <cmath>
using namespace std;
int num[100010];
int square[10010][10010];
bool cmp(int x, int y){
    return x > y;
}
int main(){
    int N, n, m;
    cin >> N;
    for(int i = 0; i < N; i ++) cin >> num[i];
    for(int i = sqrt(N); i >= 1; i --){
        if(i * (N / i) == N){
            n = i;
            m = N / i;
            break;
        }
    }
    sort(num, num + N, cmp);
    //m行n列
    int level = m / 2 + m % 2;
    int cnt = 0;
    for(int i = 0; i < level ; i ++){
        for(int j = i; j <= n - i - 1 && cnt < N; j ++)
            square[i][j] = num[cnt ++];
        for(int j = i + 1; j <= m - i - 2 && cnt < N; j ++)
            square[j][n - i - 1] = num[cnt ++];
        for(int j = n - i - 1; j >= i && cnt < N; j--)
            square[m - i - 1][j] = num[cnt ++];
        for(int j = m - i - 2; j >= i + 1 && cnt < N; j --)
            square[j][i] = num[cnt ++];
    }
    for(int i = 0; i < m; i++){
        for(int j = 0; j < n - 1; j ++){
            cout << square[i][j] << " ";
        }
        cout << square[i][n - 1] << endl;
    }
    
    return 0;
}