LeetCode118. 杨辉三角

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

在杨辉三角中,每个数是它左上方和右上方的数的和。

示例:

输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]

import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
    public ArrayList<ArrayList<Integer>> generate(int numRows) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if (numRows == 0) {
            return res;
        }
        if (numRows == 1) {
             res.add(new ArrayList<>(Arrays.asList(1)));
            return res;
        }
        if (numRows == 2) {
            res.add(new ArrayList<>(Arrays.asList(1)));
            res.add(new ArrayList<>(Arrays.asList(1,1)));
            return res;
        }
        res.add(new ArrayList<>(Arrays.asList(1)));
        res.add(new ArrayList<>(Arrays.asList(1,1)));
        for (int i = 2; i < numRows; i++) {
            ArrayList<Integer> last = res.get(i -1);
            ArrayList<Integer> cur = new ArrayList<>();
            cur.add(1);
            for (int j = 1; j < i; j++) {
                cur.add(last.get(j) + last.get(j -1));
            }
            cur.add(1);
            res.add(cur);
        }
        return res;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_31617121/article/details/80458484