LeetCode-杨辉三角

LeetCode-杨辉三角

题目描述

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

示例:
LeetCode-杨辉三角_第1张图片

思路说明

每一行的数字个数就是所在行数,每一行的数据中首尾两个数是1,同时上一行中相邻两个数字的和就是下一行中的中间数字,而numRows为1或者为2时是特例,所以numRows为1时,返回[[1]],numRows为2时,返回[[1],[1,1]];而当numRows大于2时,先计算上一行两两之和,然后计算当前数据时,先加1,再拼接上上一行的计算结果,再拼接上1就可以了。
欢迎大家批评指正。

代码实现

class Solution(object):
    def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        if numRows == 1:
            return [[1]]
        elif numRows == 2:
            return [[1], [1, 1]]
        else:
            num_list = [[1], [1, 1]]
            for i in range(3, numRows + 1):
                last_level = num_list[-1]
                last_bi_sum = []
                temp = []
                for j in range(1, len(last_level)):
                    last_bi_sum.append(last_level[j - 1] + last_level[j])
                temp.append(1)
                temp.extend(last_bi_sum)
                temp.append(1)
                num_list.append(temp)
            return num_list

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle/

你可能感兴趣的:(python,LeetCode,leetcode,python)