给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
-
示例:
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
评论区大神太强了。。。
解题思路:
观察一下规律,发现当前一行只比上一行多了一个元素,最最关键的一点:本行元素等于上一行元素往后错一位再和上一行元素自身逐个相加:
因此我们只要对最后一行单独处理:最后一行首、尾分别添加一个零然后对应位置求和就可以得到新的一行,思路上比较清晰,占用的时间、空间复杂度也都还挺好
class Solution:
def generate(self, numRows: int) -> List[List[int]]:
if numRows == 0:
return []
result = [[1]]
while len(result) < numRows:
result.append([a + b for a, b in zip(result[-1] + [0], [0] + result[-1])])
return result
- 两个列表直接相加即可实现append的效果
- zip函数也用的很妙!
#举例说明zip函数用法
a = [1, 2, 3]
b = [4, 5, 6]
c = [4, 5, 6, 7, 8]
a_b_zip = zip(a, b) # 打包为元组的列表,而且元素个数与最短的列表一致
print("type of a_b_zip is %s" % type(a_b_zip)) # 输出zip函数的返回对象类型
a_b_zip = list(a_b_zip) # 因为zip函数返回一个zip类型对象,所以需要转换为list类型
print(a_b_zip)
print("------------------------------------------")
a_c_zip = zip(a, c)
a_c_zip = list(a_c_zip)
print(a_c_zip)
输出结果:
type of a_b_zip is <class 'zip'>
[(1, 4), (2, 5), (3, 6)]
------------------------------------------
[(1, 4), (2, 5), (3, 6)]
对于 zip() 函数返回的 zip 对象,既可以通过遍历提取其存储的元组,也可以通过调用 list()函数将 zip() 对象强制转换成列表。