作者:看那个码农
公众号:看那个码农
本题来源于「LeetCode系列」118. 杨辉三角
题目描述:
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
提示:
- 0 <= key <= 10^6
- 最多调用 10^4 次 add、remove 和 contains 。
代码分析
从题意和杨辉三角的特性
- 每个数是它左上方和右上方的数的和。
所以我们可以创建一个空数组res
接着采用递归的方式,从上往下第一行开始不断生成元素均为1的新数组tmp
同时根据杨辉三角,每个数是它左上方和右上方的数的和的联系,不断更新tmp数组的值
将每一个更新数字后的数组tmp保存到空数组res中
最后生成完全体杨辉三角res
完整代码为:
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
res = []
for i in range(numRows):
tmp = [1] * (i + 1)
for j in range(i + 1):
if 0 < j < i:
tmp[j] = res[i - 1][j - 1] + res[i - 1][j]
res.append(tmp)
return res
如果你觉得这篇内容对你有帮助的话:
1、点赞支持下吧,让更多的人也能看到这篇内容
2、关注公众号:看那个码农,我们一起学习一起进步。
本文正在参与「掘金 2021 春招闯关活动」, 点击查看 活动详情