原题链接:https://leetcode.com/problems/pascals-triangle-ii/description/
题意:这道题和118题类似,但是这道题要求的输出是第k行的输出即可。
思路:可以利用杨辉三角一个很重要的特性快速得到结果,那就是一行的数组可以有上一行的数组前后分别添加0 相加得到
题解:
class Solution(object):
def getRow(self, rowIndex):
"""
:type rowIndex: int
:rtype: List[int]
"""
"""
Time Complexity:O(K^2)
Space Complexity:O(K)
"""
row = [1]
for _ in range(rowIndex):
row = [x + y for x, y in zip([0] + row, row + [0])]
return row