118. 杨辉三角

2022年10月26日

一、问题描述

给定一个非负整数 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 <= numRows <= 30

二、问题分析

• 算法思路如下：使用数学方法，逐行地计算出杨辉三角。
  

三、解题代码

class Solution(object):
    def generate(self, numRows):
        """
        :type numRows: int
        :rtype: List[List[int]]
        """
        res = []
        for i in range(1, numRows+1):
            row = []
            if i == 1:
                row.append(1)
            else:
                for j in range(i):
                    if j == 0 or j == i-1:
                        row.append(1)
                    else:
                        row.append(res[i-2][j-1]+res[i-2][j])
            res.append(row)
        
        return res

总结

使用数学方法，逐行地计算出杨辉三角。