给定一个非负索引 rowIndex
,返回「杨辉三角」的第 rowIndex
行。
在「杨辉三角」中,每个数是它左上方和右上方的数的和。
示例 1:
输入: rowIndex = 3
输出: [1,3,3,1]
示例 2:
输入: rowIndex = 0
输出: [1]
示例 3:
输入: rowIndex = 1
输出: [1,1]
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
res = list()
if rowIndex==0:
res=[1]
return res
else:
for i in range(rowIndex+1):
row = list()
for j in range(0, i+1):
if j==0 or j==i:
row.append(1)
else:
row.append(res[i-1][j-1]+res[i-1][j])
res.append(row)
return res[rowIndex]
这道跟昨天的类似,不过需要注意的是要返回的是rowIndex
行,所以要求出rowIndex
行的数再返回,也就是rowIndex+1