118. 杨辉三角

题目描述

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

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

示例:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/pascals-triangle
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

我的思路

除了第一、二行，从第三行开始的每一行都与上一行有关，第2个元素 = 上一行的第1个元素 + 上一行的第2个元素，第3个元素 = 上一行的第2个元素 + 上一行的第3个元素，以此类推，当无法进行上述操作时，首位补1，就得到了当前行的所有元素。看上面的动图一目了然，程序实现起来也不难。

写程序时，把结果写成下面这种形式，看起来更方便。

[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]

我的程序（Java）

import java.util.*;
class Solution {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new LinkedList<>();
        List<Integer> row = new LinkedList<>();
        if(numRows == 0)    return res;    // 返回空列表
                
        row.add(1); // [1]
        res.add(new LinkedList<>(row)); // 添加第一个元素[1]
        if(numRows == 1)    return res;
        
        row.add(1); // [1,1]
        res.add(new LinkedList<>(row)); // 添加第二个元素[1,1]
        if(numRows == 2)    return res;
        
        for(int i = 2; i < numRows; i++){
            List<Integer> newRow = new LinkedList<>();
            newRow.add(1);	// 添加首尾元素
            newRow.add(1);
            List<Integer> temp = res.get(res.size()-1); // 取出目前结果res的最后一个列表
            for(int j = 1; j < i; j++){
                newRow.add(j, temp.get(j-1) + temp.get(j));
            }
            res.add(new LinkedList<>(newRow)); // 添加新的列表
        }
        return res;
    }
}

如有不当之处，欢迎读者批评指正！