根据题目进行模拟或者找规律即可;
第一眼看到题目想到可以建立一个n x n的矩阵,将数字i需要存储的索引位置按规律找到后写入,但是这个方法时间空间消耗都很大。
因此想到,既然可以按规律写出矩阵,那么为什么不按规律直接输出呢?
观察案例可知:

  1. 总行数为n行的蛇形矩阵,第x行(索引以0开始)的元素个数为n - x
  2. 第x行的起始元素为上一行的起始元素加x,即f(x) = f(x-1)+x 特殊情况出现在第0行,因此需要加一个判断
  3. 第x行的第p个元素为第p-1个元素加x加p加1,即f(x, p) = f(x, p-1) + x + p + 1
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = "";
        while ((s = br.readLine()) != null) {
            int n = Integer.parseInt(s.trim());
            int start = 1;
            for (int row = 0; row < n; row++) {
                if (row > 0) {
                    start = start + row;
                }
                int current = start;
                System.out.print(current);
                for (int order = 1; order < n - row; order++) {
                    current += row + order + 1;
                    System.out.print(" " + current);
                }
                System.out.println();
            }
        }
        br.close();
    }
}