华为OJ076-蛇形矩阵

Posted 2020-07-07 Wang-Junchao

【华为OJ】【算法总篇章】

---

【华为OJ】【076-蛇形矩阵】

【工程下载】

---

题目描述

题目说明
    蛇形矩阵是由1开始的自然数依次排列成的一个矩阵上三角形。
样例输入
    5
样例输出
    1 3 6 10 15
    2 5 9 14
    4 8 13
    7 12
    11

输入描述

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

输出描述

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

输入例子

4

输出例子

1 3 6 10
2 5 9
4 8
7

算法实现

import java.util.Scanner;

/**
 * Author: 王俊超
 * Date: 2016-01-03 13:58
 * Declaration: All Rights Reserved !!!
 */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
//        Scanner scanner = new Scanner(Main.class.getClassLoader().getResourceAsStream("data.txt"));
        while (scanner.hasNext()) {
            int n = scanner.nextInt();
            System.out.print(serpentineMatrix(n));
        }

        scanner.close();
    }

    /**
     * 蛇形矩阵
     *
     * @param n 矩阵行数
     * @return 矩阵输出结果
     */
    private static String serpentineMatrix(int n) {

        StringBuilder builder = new StringBuilder();


        for (int i = 1; i <= n; i++) {
            // 每一行的第一个元素是(i-1)*i/2+1
            // 每一行的元素个数是n-i
            // 初始间隔是i+1，之后每一个间隔比上一个间隔多1，之后的每个元素是前一个元素加上间隔
            for (int j = 1, start = (i - 1) * i / 2 + 1, step = i + 1; j <= n - i + 1; j++, start += step, step++) {
                builder.append(start).append(' ');
            }

            // 设置换行符
            builder.setCharAt(builder.length()-1, '\\n');

        }


        return builder.toString();
    }
}