[Leetcode] 0119. 杨辉三角 II

目录

119. 杨辉三角 II
  题目描述
  解法

119. 杨辉三角 II

题目描述

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

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

 

示例 1:

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

示例 2:

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

示例 3:

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

 

提示:

• 0 <= rowIndex <= 33

 

进阶：

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

解法

该方法是常见的方法，即按照新建一个二维数组 res[i][j] ，数组的每一行 res[i] 代表了杨辉三角的第 i 行的所有元素， res[i][j] 表示杨辉三角的第 i 行第 j 列的元素。。

由下面的图我们可以看出： $res[i][j] = res[i - 1][j - 1] + res[i - 1][j]$。

时间复杂度：$O(\textit{rowIndex}^2)$。

Python3

class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        res = [[1 for j in range(i + 1)] for i in range(rowIndex + 1)]
        for i in range(2, rowIndex + 1):
            for j in range(1, i):
                res[i][j] = res[i - 1][j - 1] + res[i - 1][j]
        return res[rowIndex]

C++

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<vector<int>> C(rowIndex+1);
        for(int i=0;i<rowIndex+1;++i){
            C[i].resize(i+1);
            C[i][0] = C[i][i] = 1;
            for(int j=1;j<i;++j){
                C[i][j]  = C[i-1][j-1] +C[i-1][j];
            }
        }
        return C[rowIndex];
    }
};

作者：野哥李
微信公众号：AI算法学习社
欢迎任何形式的转载，但请务必注明出处。
限于本人水平，如果文章和代码有表述不当之处，还请不吝赐教。
本文章不做任何商业用途，仅作为自学所用，文章后面会有参考链接，我可能会复制原作者的话，如果介意，我会修改或者删除。