力扣LeetCode第一百一十八题 杨辉三角

数据结构（九十四）

学习数据结构与算法过程中的心得体会以及知识点的整理，方便我自己查找，也希望可以和大家一起交流。

—— 杨辉三角 ——

1.题目描述

给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。在「杨辉三角」中，每个数是它左上方和右上方的数的和。

示例：

示例 1:
输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

示例 2:
输入: numRows = 1
输出: [[1]]

2.代码

python

class Solution(object):
    def generate(self, numRows):
        if numRows == 0: 
            return []
        res = [[1]]
        while len(res) < numRows:
            newRow = [a+b for a, b in zip([0]+res[-1], res[-1]+[0])]
            res.append(newRow)      
        return newRow

思路：通过杨辉三角的规律来解决问题，前一行首位加 0，记为 a 行，前一行末位加 0，记为 b 行。a 和 b 按照位次相加，即为下一行结果。原本的实现方法太过繁琐，在题解中找到了简单的实现方法，通过zip函数实现。