描述
给定一个非负整数 numRows,生成杨辉三角的前 numRows 行。
在杨辉三角中,每个数是它左上方和右上方的数的和。
示例
输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]
我
if not numRows:
return []
p = [1]
r = [p]
i = 1
while i < numRows:
i+=1
s = [0] * i
p = [0] + p + [0]
for j in range(i):
s[j] = p[j] + p[j+1]
r.append(s)
p = s
return r
通过了,终于暂时没有树的题了,松了一口气。大神们都用lambda以及map()做的。。。emmm明天来研究。
总结:
1. lambda表达式的用法
+ lambda的主体是一个表达式
+ 例子:
>>> f = lambda x,y: x+y
>>> f(1,2)
3
>>> a = [1,2,3]
>>> b = [4,5,6]
>>> c = [7,8,9]
>>> d = [4,5]
>>> list(zip(a,b,c))
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
>>> list(zip(a,d))
[(1, 4), (2, 5)]
最常应用的地方是矩阵行列互换。
>>> a = [[1,2,3],[4,5,6],[7,8,9]]
>>> list(map(list, zip(*a)))
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
>>>
>>> list(map(lambda x:x%2, range(4)))
[0, 1, 0, 1]
+多个seq,一个返回值
>>> list(map(lambda x,y: x+y,[1,2],[3,4]))
[4, 6]
+多个seq,多个返回值组成一个元组
>>> list(map(lambda x,y: (x+y, x*y), [1,2,3],[4,5,6]))
[(5, 4), (7, 10), (9, 18)]
>>> from functools import reduce
>>> reduce(lambda x,y:x*y, [1,2,3,4,5])
120
2倍阶乘,初始参数为2。
>>> reduce(lambda x,y:x*y, [1,2,3,4,5],2)
240
5.filter()函数
filter()函数用于过滤序列。和map()类似,filter()也接收一个函数和一个序列。和map()不同的是,filter()把传入的函数依次作用于每个元素,然后根据返回值是True还是False决定保留还是丢弃该元素。
例子:删掉偶数
>>> list(filter(lambda x: x%2 == 1, [1,2,3,4,5]))
[1, 3, 5]
好,回到正题。
优化后的代码
r = [[1]]
for i in range(1,numRows):
r.append(list(map(lambda x,y:x+y, [0]+r[-1],r[-1]+[0])))
return r[:numRows]#这里太强了