杨辉三角
描述
简单
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
解题
相当于把三角的数据进行左对齐,得
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
除第1、2行和每行的左右两边的数据外,其余的数据等于左上方的数据加上上方的数据
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
if numRows == 0:
return []
elif numRows == 1:
return [[1]]
elif numRows == 2:
return [[1], [1, 1]]
trangle = [[1], [1, 1]]
for i in range(2, numRows):
temp = [1 for _ in range(i+1)]
# 左右两边的数据不进行计算
for j in range(1, len(temp)-1):
temp[j] = trangle[i-1][j-1] + trangle[i-1][j]
trangle.append(temp)
return trangle
杨辉三角II
描述
简单
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
解题
按照上面的解法迭代到k层再返回也是可以的,但空间复杂度就不能满足要求了
首先申请所需的空间,也是多次迭代更新
每次更新时从后往前更新,这样就不会影响下一个数的更新
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
row = [1 for i in range(rowIndex+1)]
for i in range(rowIndex+1):
# 每次每层左右两端的数都不更新
for j in range(i-1, 0, -1):
row[j] = row[j] + row[j-1]
return row