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];
}
};