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
总结
使用数学方法,逐行地计算出杨辉三角。