PTA 1050 螺旋矩阵(详解+代码)

啊，二刷才借鉴别人的代码，耐心的研究，我的思路是这样的：
1.先求出m、n，其实也就是根号N可以得出公约数之一，如果N对这个公约数取余！=0，那么说明不是整除的，所以我们再n–，就可以得出比m小的n了，n已知，m=N/n;
2.将给的数字存到数组，然后升序，方便后面的降序。
3.创建一个二维数组，他的俩个下标进行对应的操作，用switch来判断方向，0是向右，1是向下，2是向左，3是向上。

package com.zcl.pta2;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
import java.util.ArrayList;
import java.util.Arrays;

/**
 * @author 曾小白
 * @create 2020/4/30 13:15
 * @desc 1.求出mn
 * 2.数字升序排序，后面方便使用
 * 3.创建二维数组按左下左上这样的方向输出，一直到所有数输出完
 **/
public class Pta_1050 {
    public static void main(String[] args) throws IOException {
        BufferedReader buf = new BufferedReader(new InputStreamReader(System.in));
        StreamTokenizer st = new StreamTokenizer(buf);
        st.nextToken();
        int N = (int) st.nval;
        int[] arr = new int[N];
        for (int i = 0; i < N; i++) {
            st.nextToken();
            int val = (int) st.nval;
            arr[i] = val;
        }
        Arrays.sort(arr);//升序
        int n = (int) Math.sqrt(N);
        while (N % n != 0) {
            n--;
        }
        int m = N / n;
        int[][] two_dimension = new int[m][n];//Y X
        int direction = 0;//方向 0右 1下 2左 3上
        int k1 = 0, k2 = 0;//俩个维度
        for (int i = N - 1; i >= 0; i--) {//降序
            two_dimension[k1][k2] = arr[i];//Y X
            switch (direction) {
                case 0://右
                    if ((k2 + 1) < n && two_dimension[k1][k2 + 1] == 0) {//未越界且没有值
                        k2++;
                    } else {
                        k1++;
                        direction = 1;
                    }
                    break;
                case 1://下
                    if ((k1 + 1) < m && two_dimension[k1 + 1][k2] == 0) {//未越界且没有值
                        k1++;
                    } else {
                        k2--;
                        direction = 2;
                    }
                    break;
                case 2://左
                    if ((k2 - 1) >= 0 && two_dimension[k1][k2 - 1] == 0) {//未越界且没有值
                        k2--;
                    } else {
                        k1--;
                        direction = 3;
                    }
                    break;
                case 3://上
                    if ((k1 - 1) >= 0 && two_dimension[k1 - 1][k2] == 0) {//未越界且没有值
                        k1--;
                    } else {
                        k2++;
                        direction = 0;
                    }
                    break;
            }
        }
        for (int i = 0; i < m; i++) {
            StringBuffer bf = new StringBuffer();
            for (int j = 0; j < n; j++) {
                if (j == 0) {
                    bf.append(two_dimension[i][j]);
                } else {
                    bf.append(" " + two_dimension[i][j]);
                }
            }
            System.out.println(bf.toString().trim());
        }
    }
}