1.题目描述
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
2.解题思路
解法一:按照定义解题
解法二:取巧法
观察一下规律,发现当前一行只比上一行多了一个元素,最最关键的一点:本行元素等于上一行元素往后错一位再逐个相加。
3.程序代码
解法一:
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
if numRows==0:
pascal=[]
if numRows==1:
pascal=[[1]]
if numRows==2:
pascal=[[1],[1,1]]
if numRows>2:
pascal=[[1],[1,1]]
for i in range(1,numRows-1):
pascal1=[1,1]
for j in range(1,i+1):
pascal1.insert(j,pascal[-1][j-1]+pascal[-1][j])
pascal.append(pascal1)
return pascal
时间:40ms;空间:13.7MB
解法二:
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
if numRows==0:
pascal=[]
if numRows==1:
pascal=[[1]]
if numRows>1:
pascal=[[1]]
for i in range(numRows-1):
temp=[a+b for a,b in zip([0]+pascal[-1],pascal[-1]+[0])]
pascal.append(temp)
return pascal
时间:36ms;空间:13.7MB