LeetCode 118.杨辉三角(python)

题目链接

题目描述:
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。

示例:

输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
解题思路:
每一行元素的个数等于行数,将每一行的元素初始化为1,从第三行开始,除了第一个元素和最后一个元素,每个元素是它左上方和右上方的数的和(也就是上一行的索引相同的位置和索引减一位置的元素的和)

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        res=[]
        for i in range(0,numRows):
            t=[1]*(i+1)#将每一行的元素初始化为1
            if i>1:#从第三行开始
                for j in range(1,i):#除了第一个元素和最后一个元素
                    t[j]=res[i-1][j]+res[i-1][j-1] #每个元素是它左上方和右上方的数的和
            res.append(t)
        return res

猜你喜欢

转载自blog.csdn.net/weixin_44740082/article/details/88796529