力扣算法之杨辉三角(动态规划求解)

1.题目描述:

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

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

2.

示例 1:

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

输入: numRows = 1
输出: [[1]]

3.python代码:

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        ans = [[1]]                                              # 初始化杨辉三角,即第一行为 1
        for i in range(1, numRows):                              # 从第二行开始遍历
            lst = [0 for _ in range(i+1)]                        # 创建一个长度为 i+1 的列表,全部初始化为 0         
            lst[0], lst[-1] = 1, 1                                # 当前行的头和尾都为 1,因此初始化为 1
            for j in range(1,i):                                  # 遍历中间元素,计算值
                lst[j] = ans[i-1][j-1] + ans[i-1][j]             # 当前元素等于上一行相邻两个元素之和,因此获取上一行对应位置的元素值
            ans.append(lst)                                       # 将当前行添加到 ans 列表中
        return ans                                                 # 返回生成的杨辉三角列表

猜你喜欢

转载自blog.csdn.net/m0_62690134/article/details/130114726
今日推荐