「LeetCode系列」杨辉三角问题|刷题打卡

120 阅读1分钟

作者:看那个码农

公众号:看那个码农

本题来源于「LeetCode系列」118. 杨辉三角

题目描述:

给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。

在杨辉三角中,每个数是它左上方和右上方的数的和。

示例:
输入: 5
输出:
[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]

提示:

  1. 0 <= key <= 10^6
  2. 最多调用 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 春招闯关活动」, 点击查看 活动详情