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] #返回想要的目标行
学习过程中,欢迎指正。