PTA 螺旋矩阵（java）

1050 螺旋矩阵 (25分)

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

样例解答：

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
 public static void main(String[] args) throws Exception{
  BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
  int N = Integer.parseInt(bf.readLine());
  int m=0,n=0;
  for(int i=1;i<=Math.sqrt(N);i++) {
   if(N%i==0&&i<=N/i) {
    m=N/i;
    n=i;
   }
  }
  //System.out.print(m+" "+n);
  String[] s = bf.readLine().split(" ");
  int[] a = new int[N];
  for(int i=0;i<N;i++) {
   a[i]=Integer.parseInt(s[i]);
  }
  Arrays.sort(a);
  int num = 1;
  int[][] sum = new int[m][n];
  for(int i=0;num<=N;i++) {
   for(int j=i;j<=n-1-i&&num<=N;j++) {
    sum[i][j]=a[N-num];
    num++;
   }
   for(int j=i+1;j<=m-2-i&&num<=N;j++) {
    sum[j][n-1-i]=a[N-num];
    num++;
   }
   for(int j=n-i-1;j>=i&&num<=N;j--) {
    sum[m-1-i][j]=a[N-num];
    num++;
   }
   for(int j=m-2-i;j>=i+1&&num<=N;j--) {
    sum[j][i]=a[N-num];
    num++;
   }
  }
  for(int i=0;i<m;i++) {
   for(int j=0;j<n;j++) {
    System.out.print(sum[i][j]);
    if(j!=n-1) System.out.print(" ");
   }
   if(i!=m-1) System.out.println();
  }
 }
}