leetcode 118. 杨辉三角 119.杨辉三角2

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

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

示例 1:

输入: numRows = 5

输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

示例 2:

输入: numRows = 1

输出: [[1]]

通过观察本题，可以得知每一行的第一个数和最后一个数为1（除第一行外），其他地方的数字是左上方和右上方的数字之和，在列表中体现为是正上方的数字和左上方的数字之和

1

11

121

1331

也就是编号为[i][j]的数字等于编号为[i-1][j]+[i-1][j-1]的数字之和

从而我们可以通过这个规律写出杨辉三角每一行的数字

class Solution:
    def generate(self, numRows):
        ret = []#用来记录杨辉三角最后的结果
        for i in range (numRows):
            row = []#记录每一行的元素
            for j in range(i+1):
                if j == 0 or j == i:
                        row.append(1)#记录每一行第一个和最后一个元素
                else:
                    row.append(ret[i-1][j]+ret[i-1][j-1])#计算出每一行其他元素
            ret.append(row)
        return ret

119. 杨辉三角 II

给定一个非负索引 rowIndex，返回「杨辉三角」的第 rowIndex 行。

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

示例 1:

输入: rowIndex = 3

输出: [1,3,3,1]

示例 2:

输入: rowIndex = 0

输出: [1]

示例 3:

输入: rowIndex = 1

输出: [1,1]

思路基本和杨辉三角一样，把需要求的那一行作为列表的最后一项输出即可

class Solution:
    def getRow(self, numRows) :
        ret = list()
        for i in range (numRows+1):
            row = list()
            for j in range(i+1):
                if j == 0 or j == i:
                        row.append(1)
                else:
                    row.append(ret[i-1][j]+ret[i-1][j-1])
            ret.append(row)
        return ret[-1]