一、刷题内容
原题链接
https://leetcode-cn.com/problems/pascals-triangle/
内容描述
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
二、解题方案
1.方法一
动态规划,需要前一行来构造当前行。时间复杂度O(n2),空间复杂度O(n2)
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
摆正后想,每个数字是它前一行的[当前下标] + 前一行[当前下标-1] 的 值
class Solution:
def generate(self, num_rows):
res = []
for i in range(num_rows): # 一行行的循环
tmp_row = [None for _ in range(i + 1)] # 构造好每一行临时列表
tmp_row[0], tmp_row[-1] = 1, 1 # 每一行的头和尾都肯定是1
for j in range(1, len(tmp_row) - 1): # 跳过头和尾
tmp_row[j] = res[i - 1][j] + res[i - 1][j - 1] # 是前一行的 j 和 j-1(代入)
res.append(tmp_row)
return res
2.方法二
class Solution:
def generate(self, numRows):
res = []
for i in range(numRows):
now = [1] * (i + 1) # 构造当前行的列表
if i >= 2: # 跳过前两行
for n in range(1, i):
now[n] = pre[n - 1] + pre[n]
res.append(now)
pre = now # 下次循环的pre是当前循环的now
return res
3.方法三:
米奇妙妙屋
妙到家了
class Solution:
def generate(self, numRows):
l1 = [1]
l2 = []
n = 0
while n < numRows:
l2.append(l1)
l1 = [sum(t) for t in zip([0] + l1, l1 + [0])]
n += 1
return l2