题目
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
方法1:滚动数组
1、将杨辉三角(到题目输入的数为止)存入一个列表list()中
2、用enumerate()函数遍历杨辉三角,再输出。
3、滚动数组实现思路
class Solution(object):
def getRow(self, rowIndex):
"""
:type rowIndex: int
:rtype: List[int]
"""
res = [1] * (rowIndex + 1)
for i in range(2, rowIndex + 1):
for j in range(i - 1, 0, -1):
res[j] += res[j - 1]
return res
作者:fuxuemingzhu
链接:https://leetcode-cn.com/problems/pascals-triangle-ii/solution/gun-dong-shu-zu-de-dong-hua-tu-jie-you-h-rw5n/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
方法2:暴力
Python实现
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
out=list()
for i in range(rowIndex+1):
var=list()
for j in range(i+1):
if j==0 or j==i:
var.append(1)
else:
var.append(out[i-1][j-1]+out[i-1][j])
out.append(var)
return out[-1]
Java实现
class Solution {
public List<Integer> getRow(int rowIndex) {
List<List<Integer>> list = new ArrayList<>();
for (int i = 0; i <= rowIndex; i++) {
// List<Integer>类存储每行的值
List<Integer> row = new ArrayList<>();
for (int j = 0; j <= i; j++) {
// 边界值赋1
if (j == 0 || j == i) {
row.add(1);
} else {
row.add(list.get(i - 1).get(j - 1) + list.get(i - 1).get(j));
}
}
list.add(row);
}
return list.get(rowIndex);
}
}