PAT (Basic Level) Practise （中文） 1050. 螺旋矩阵(25)

1050. 螺旋矩阵(25)

时间限制

150 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

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

输入格式：

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

输出格式：

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

//部分正确 18分
//思路 先获取那个数组排序，然后根据数组长度算m和n
//创建二维数组 改变下标做填充
import java.util.Arrays;
import java.util.Scanner;

public class Main {
  public static void main(String[] args) {
     Scanner input=new Scanner(System.in);
     int N=input.nextInt();
    
     int input_arr[]=new int[N];
     for (int i = 0; i < N; i++) {
       input_arr[i]=input.nextInt();
     }
//    int N = 12;
//    int input_arr[] = { 37, 76, 20, 98, 76, 42, 53, 95, 60, 81, 58, 93 };
    Arrays.sort(input_arr);

    int n = (int) Math.sqrt(N);
    int m = N / n;
    int matrix[][] = new int[m][n];
    int index = N - 1;
    int i = 0, j = 0;
    while (index >= 0) {
      // right.
      for (; index >= 0 && j < n && matrix[i][j] == 0; j++) {
        matrix[i][j] = input_arr[index--];
      }
      i++;
      j--;

      // down.
      for (; index >= 0 && i < m && matrix[i][j] == 0; i++) {
        matrix[i][j] = input_arr[index--];
      }

      i--;
      j--;

      // left.
      for (; index >= 0 && j >= 0 && matrix[i][j] == 0; j--) {
        matrix[i][j] = input_arr[index--];
      }
      i--;
      j++;

      // up.
      for (; index >= 0 && i >= 0 && matrix[i][j] == 0; i--) {
        matrix[i][j] = input_arr[index--];
      }
      i++;
      j++;
    }
    for (int p = 0; p < m; p++) {
      System.out.print(matrix[p][0]);
      for (int q = 1; q < n; q++) {
        System.out.print(" " + matrix[p][q]);
      }

      System.out.println("");
    }

  }
}