Python学习日记——杨辉三角、杨辉三角II

Python学习日记013——杨辉三角、杨辉三角II

题目来源：LeetCode题库——杨辉三角、杨辉三角II
①给定一个非负整数 numRows，生成杨辉三角的前 numRows 行。
②给定一个非负索引 k，其中 k ≤ 33，返回杨辉三角的第 k 行。

题①：Example1：
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
链接：https://leetcode-cn.com/problems/pascals-triangle

题②：Example2：
输入: 3
输出: [1,3,3,1]
链接：https://leetcode-cn.com/problems/pascals-triangle-ii

构建杨辉三角
通过观察杨辉三角，可以得到它具有以下几个特点：
①每行的元素个数等于行数。即第1行有1个元素，第2行有2个元素，第3行有3个元素……
②每行开头和结尾的数字均为1。
③从第三行开始，除开头和结尾的数字外，每行第 m 个数字等于前一行第 m-1 个数字和第 m 个数字之和。
我们可以从杨辉三角的这三个特点入手来解决这个问题：
根据特点①，我们可以先得出一个初始数组，这个数组元素全为 0 ，且元素个数等于行数的数组。
根据特点②，我们可以将每行的第一个和末尾元素设置为 1 。这样第一行已经确定，为 [1] 。第二行也随之确定，为 [1,1] 。
根据特点③，我们可以从第三行开始，遍历每一行内的元素，因为首尾两个元素在第二步中已被设置为 1 ，那么我们只需要修改仍为 0 的元素。当搜索到元素为0时，其位置上正确的值应为 前一行该位置元素的值 与 前一行该元素前一位的值 的和。
具体实现过程如下：

class Solution:
    def generate(self, numRows: int) -> List[List[int]]:
        tri = [[0]*i for i in range(1,numRows+1)]  #初始构建元素全部为 0 的三角形阵列

        for i in range(numRows):
            tri[i][0] = tri[i][-1] = 1   #将边界值设置为1
            
        for i in range(2,numRows):    #对于其他元素，按照杨辉三角的计算法则求得结果
            for j in range(1,i):
                if tri[i][j] == 0:
                    tri[i][j] = tri[i-1][j-1] + tri[i-1][j]
        return tri
        

返回杨辉三角的第k行
对于这个问题，可以以构建杨辉三角为基础，先构建出杨辉三角，然后返回所想要的目标行的内容。

class Solution:
    def getRow(self, rowIndex: int) -> List[int]:
        tri = [[0]*i for i in range(1,rowIndex+2)]
        
        for i in range(rowIndex+1):
            tri[i][0] = tri[i][-1] = 1
        
        for i in range(2,rowIndex+1):
            for j in range(1,i+1):
                if tri[i][j] == 0:
                    tri[i][j] = tri[i-1][j-1] + tri[i-1][j]
        
        return tri[rowIndex]  #返回想要的目标行
        

学习过程中，欢迎指正。