根据题目进行模拟或者找规律即可;
第一眼看到题目想到可以建立一个n x n的矩阵,将数字i需要存储的索引位置按规律找到后写入,但是这个方法时间空间消耗都很大。
因此想到,既然可以按规律写出矩阵,那么为什么不按规律直接输出呢?
观察案例可知:
- 总行数为n行的蛇形矩阵,第x行(索引以0开始)的元素个数为n - x
- 第x行的起始元素为上一行的起始元素加x,即
f(x) = f(x-1)+x
特殊情况出现在第0行,因此需要加一个判断 - 第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(); } }