[Leetcode] 0119. 杨辉三角 II

杨辉三角 II"/>

[Leetcode] 0119. 杨辉三角 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];}
};