PAT 1050 螺旋矩阵 (JAVA)

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

我的思路： 感觉是一道容易让人厌烦的题目，在leetcode有一道类似的题目，那道是 按顺时针打印矩阵。总之核心实际上是控制四个边界（上下左右），没有想的那么烦。（有两个测试点超时）

import java.util.*;
import java.io.*;
public class Main{
    public static void main(String[]args)throws IOException{
        BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
        List<Integer>list=new ArrayList<>();
        int n=Integer.parseInt(bf.readLine());
        int y=(int)Math.sqrt(n);
        while(n%y!=0)
            --y;
        int x=n/y;
        int[][]matrix=new int[x][y];
        //
        String []strs=bf.readLine().split(" ");
        for(int i=0;i<n;++i) {
        	list.add(Integer.parseInt(strs[i]));
        }
        Collections.sort(list);
        //核心代码
        int l=0,u=0,r=y-1,d=x-1;
        int cnt=n-1;
        while(true){
            for(int i=l;i<=r;++i) matrix[l][i]=list.get(cnt--);
            if(++u>d) break;
            for(int i=u;i<=d;++i) matrix[i][r]=list.get(cnt--);
            if(--r<l) break;
            for(int i=r;i>=l;--i) matrix[d][i]=list.get(cnt--);
            if(--d<u) break;
            for(int i=d;i>=u;--i) matrix[i][l]=list.get(cnt--);
            if(++l>r) break;
        }
        //
        for(int i=0;i<x;++i) {
        	for(int j=0;j<y;++j){
        		if(j!=0)
        			System.out.print(" ");
        		System.out.print(matrix[i][j]);
        	}
        	System.out.println();
        }
    }
}

java能AC的大佬救我