Java实现蛇形矩阵

需求：

蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。
例如，当输入5时，应该输出的三角形为：
1 3 6 10 15
2 5 9 14
4 8 13
7 12
11

实现：

  public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    System.out.print("请输入数字：");
    int n = scanner.nextInt();

    int[][] arr = new int[n][n];
    int index = 1;

    //索引和为i
    for (int i = 0; i < n; i++) {
      for (int j = 0; j <= i; j++) {
        arr[i - j][j] = index;
        index++;
      }
    }

    for (int[] ints : arr) {
      for (int i : ints) {
        if (i == 0) {
          continue;
        }
        System.out.print(i + " ");
      }
      System.out.println();
    }
  }