杨辉三角java实现

杨辉三角

题目描述

给定一个非负的整数numRows，生成杨辉三角的钱numRows行。

e.g. 输入：5
输出：
[1]
[1, 1]
[1, 2, 1]
[1, 3, 3, 1]
[1, 4, 6, 4, 1]

分析

规律：
1.三角形的两边都是1
2.当前元素为：斜上方和上方元素之和

代码

public class YangHuiTriangle {
    public List<List<Integer>> generate(int numRows) {
        if (numRows <= 0) {
            return new ArrayList<>();
        }
        ArrayList<List<Integer>> list = new ArrayList<>();
        for (int i = 0; i < numRows; i++) {
            List<Integer> tmp = new ArrayList<>();
            for (int j = 0; j <= i; j++) {
                if (j == 0 || j == i) {
                    tmp.add(1);
                } else {
                    //获得上方和斜上方数字之和
                    tmp.add(list.get(i - 1).get(j) + list.get(i - 1).get(j - 1));
                }
            }
            list.add(tmp);
        }
        return list;
    }

    public void test() {
        List<List<Integer>> res = generate(5);
        for (int i = 0; i < res.size(); i++) {
            System.out.println(res.get(i).toString());
        }
    }

    public static void main(String[] args) {
        YangHuiTriangle t = new YangHuiTriangle();
        t.test();
    }
}