HJ35 蛇形矩阵

HJ35 蛇形矩阵

描述

蛇形矩阵是由 1 开始的自然数依次排列成的一个矩阵上三角形。

例如，当输入 5 时，应该输出的三角形为：
1 3 6 10 15
2 5 9 14
4 8 13
7 12
11

输入描述：

输入正整数 N（N 不大于 100）

输出描述：

输出一个 N 行的蛇形矩阵。

示例 1

输入：
4
输出：
1 3 6 10
2 5 9
4 8
7

题解

    第一行: [1 3 6 10]
    第二行: [2 5 9] => [3-1 6-1 10-1]
    去掉上一行的第一列，然后将后面的每一项分别减1得到的。
    第三行: [4 8] => [5-1 9-1]
    第四行: [7] => [8-1]

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner sc  = new Scanner(System.in);
        while (sc.hasNext()) {

            int n = sc.nextInt();
            int[] nums = new int[n];
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < n; i++) {
                if (i == 0) {
                    // 第一行
                    int add = 2;
                    nums[0] = 1;
                    for (int j = 1; j < n; j++) {
                        nums[j] = nums[j - 1] + add;
                        add++;
                    }
                } else {
                    for (int j = 1; j <= n - i; j++) {
                        nums[j - 1] = nums[j] - 1;
                    }
                }

                for (int k = 0; k < n - i; k++) {
                    sb.append(nums[k] + " ");
                }
                sb.append("\n");//换行
            }
            System.out.println(sb.toString());
        }
    }
}