力扣热题100_动态规划_118_杨辉三角

---

题目链接

118. 杨辉三角

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

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

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

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

解题思路

    # 思路 1：动态规划
    # 1. 划分阶段：按照行数进行阶段划分。
    # 2. 定义状态：定义状态 dp[i][j] 为：杨辉三角第 i 行、第 j 列位置上的值。
    # 3. 状态转移方程：根据观察，很容易得出状态转移方程为：dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j]，此时，j > 0。
    # 4. 初始条件：
        # 每一行第一列都为 1，即 dp[i][0] == 1。
        # 每一行最后一列都为 1，即 dp[i][i] ==1。
    # 5. 最终结果：根据题意和状态定义，我们将每行结果存入答案数组中，将其返回。

解题代码

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        dp = [[0] * i for i in range(1, numRows + 1)]
        
        for i in range(numRows):
            dp[i][0] = 1
            dp[i][i] = 1
        
        res = []
        for i in range(numRows):
            for j in range(i):
                if i != 0 and j != 0:
                    dp[i][j] = dp[i - 1][j - 1] + dp[i - 1][j]
            res.append(dp[i])

        return res

---

参考资料：datawhalechina