Leetcode 118. 杨辉三角

题目

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

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

示例:

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

解答

很容易理解，可以看下面代码理解下述步骤。

具体为以下几步：

1. 对第一行特殊处理，相当于初始化，初始化第一行为 1 。
2. 从第二行开始：
   1. 获得设置当前位置所需要的数据 left , right 。（注意边界条件， 如果超过边界，按 0 值处理）
   2. 计算新一行的值。使用公式 matrix[ i ] [ j ] = matrix[ i - 1 ] [ j - 1 ] + matrix[ i - 1 ] [ j ] 。
   3. 将新的一行加入结果集中。

代码

class Solution {
    public List<List<Integer>> generate(int numRows) {
        if(numRows == 0) return new ArrayList<>();
        List<List<Integer>> res = new ArrayList<>();
        res.add(new ArrayList<Integer>(Arrays.asList(1)));
        for(int i = 1; i < numRows; i ++) {
            List<Integer> row = new ArrayList<>();
            for(int j = 0; j <= i; j ++) {
                int left = j == 0 ? 0 : res.get(i - 1).get(j - 1);
                int right = j == i ? 0 : res.get(i - 1).get(j);
                row.add(left + right);
            }
            
            res.add(row);
        }
        
        return res;
    }
}

结果