题解 | #蛇形矩阵#

import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        while (in.hasNextInt()) { // 注意 while 处理多个 case
            int a = in.nextInt();
            int[][] res = new int[a][a];
            int count = 1;
            for(int i=0;i<a;i++) {
                int i2 = i;
                int j = 0;
                while(j<a && i2 >=0) {
                    res[i2][j] = count;
                    count ++;
                    j ++ ;
                    i2--;
                }
            }
            for(int i =0 ;i<a;i++) {
                for(int j=0;j<a;j++) {
                    if(res[i][j] != 0) {
                        System.out.print(res[i][j] + " ");
                    }
                }
                System.out.println();
            }
        }
    }
}