Skip to content

119.杨辉三角II

滚动数组,https://leetcode.cn/problems/pascals-triangle-ii/

给定一个非负索引 rowIndex,返回「杨辉三角」的第 rowIndex 行。

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

img

示例 1:

输入: rowIndex = 3
输出: [1,3,3,1]

示例 2:

输入: rowIndex = 0
输出: [1]

示例 3:

输入: rowIndex = 1
输出: [1,1]

提示:

  • 0 <= rowIndex <= 33

进阶:

你可以优化你的算法到 *O*(*rowIndex*) 空间复杂度吗?

滚动数组都是 简单 题了?https://leetcode.cn/problems/pascals-triangle-ii/

滚动数组不易理解,可以 https://pythontutor.com/ 看可视化执行过程。

杨辉三角形需要前一行的数据来计算当前行的数据,利用一个一维数组(即滚动数组)来保存这些数据,并随着行数的增加不断更新这个数组。

dp数组实际上代表了当前行。从当前行的末尾开始向前遍历并更新dp数组中的元素。这样做的好处是不会覆盖掉计算新值所需的旧值,从而确保了算法的正确性,同时节省了额外的存储空间。

python
class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        dp = [1] * (rowIndex + 1)
        for row in range(1, rowIndex + 1):
            for i in range(row - 1, 0, -1):
                dp[i] = dp[i - 1] + dp[i]

        return dp