杨辉三角I
题目描述
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
![46e1ee8a0fc754eb13a79ae15c9c3afb.gif](https://i-blog.csdnimg.cn/blog_migrate/3e3bfaa40df68744818500a2cc70453c.gif)
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
分析与解答
可以一行一行错位加,当然这里提供更简便的方法。
任取一行描述:[1, 2, 1] 如何得到 [1, 3, 3, 1] ?这样,[0] + [1, 2, 1] 与 [1, 2, 1] + [0] 对应位相加即可。+指的是连接哦~
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
if numRows == 0:
return []
res = [[1]]
for i in range(1, numRows):
temp1 = res[-1] + [0]
temp2 = [0] + res[-1]
res.append([temp1[i] + temp2[i] for i in range(len(temp1))])
# # 上面三行,Python大神会这样写
# res.append([x + y for x, y in zip([0] + res[-1], res[-1] + [0])])
return res
杨辉三角II
题目描述
给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 k 行。
![46e1ee8a0fc754eb13a79ae15c9c3afb.gif](https://i-blog.csdnimg.cn/blog_migrate/3e3bfaa40df68744818500a2cc70453c.gif)
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 3
输出: [1,3,3,1]
进阶:
你可以优化你的算法到 O(k) 空间复杂度吗?
分析与解答
不解释。
class Solution:
def getRow(self, rowIndex: int) -> List[int]:
cur = [1]
for _ in range(rowIndex):
cur1 = [0] + cur
cur2 = cur + [0]
cur = [x + y for x, y in zip(cur1, cur2)]
return cur