每日Leetcode—算法(12)

118.杨辉三角

输入: 5
输出:
[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]

算法一(自己写的):

def generate(self, numRows: int) -> List[List[int]]:
    if numRows == 0:     #第0行为空
        return []
    if numRows == 1:
        return [[1]]        #返回第1行
    a = [[1],[1,1]]       #前两行
    for i in range(3,numRows+1):        #从第三行开始进行循环
        b = [1]              #每一行头一个数均为1
        for j in range(2,i):           #需要加入i-2个数字  
            b.append(a[i-2][j-2]+a[i-2][j-1])    #第i行的第一个数为上一行前两个数相加
        b.append(1)       #最后一个数仍为1
        a.append(b)      #将第i行加入数列
    return a 

分析:该算法的语句较多,不够简洁美观。

算法二

def generate(self, numRows: int) -> List[List[int]]:
    l = [[1]]
    for i in range(1,numRows):           #循环numRows-1次
        l.append(list(map(lambda x,y:x+y, [0]+r[-1], r[-1]+[0])))
    return l[:numRows]       #如果numRows为0,则返回[]

分析:该算法利用了杨辉三角下一行等于上一行错格相加的特性。
高手们都使用了lambda、map等函数进行编写,写出来的算法简洁美观。下面进行从map等基础函数开始介绍。

lambda函数

lambda是匿名函数,通常是在想用一个简单的函数,但是又不想def一个新函数的情况下使用,表达式为:
lambda 形参列表 : 函数返回值表达式语句

>>>f = lambda x,y: x+y
>>>f(1,2)
3

zip()函数

zip函数是Python的一个内建函数,可以接受一系列可迭代的对象作为参数,将对象打包成一个个元组(tuple),然后返回由元组组成的列表(list)。若传入参数的长度不等,则以最短长度的对象为最终list的长度。

>>> b = [4,5,6]
>>> c = [7,8,9]
>>> d = [4,5]
>>>list(zip(a,b))
[(4,7), (5,8), (6,9)]
>>>list(zip(a,b,c))
[(4,7,4),(5,8,5)]

利用*号操作符,可以将list unzip(解压),下面是将矩阵行列互换

>>> a = [[1,2,3],[4,5,6],[7,8,9]]
>>>list(map(list, zip(*a)))
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

map()函数

map(func,seq[,seq2…])
map函数将func用于seq中的每一个元素,并用一个列表将结果返回。如果func为空,则作用等同于zip()。

>>>list(map(lambda x:x%2), range(4)))
[0,1,0,1]
>>>list(map(lambda x,y:x+y), [1,2],[3,4]))
[4,6]
>>>list(map(lambda x,y: (x+y, x*y), [1,2,3],[4,5,6]))
[(5,4),(7,10),(9,18)]

reduce函数

reduce(func,seq[,init])
reduce()函数即为化简,前两个元素进行func,结果与下一个元素再次进行func。
reduce(func,[1,2,3]) = func(func(1,2),3)
其中init可选:如存在init,则init作为第一个元素。使用reduce函数需要从functool中调用。

>>>from functools import reduce
>>>reduce(lambda x,y:x+y, [1,2,3,4,5])  #阶乘
120
>>>reduce(lambda x,y:x+y, [1,2,3,4,5], 2)   #二倍阶乘
240

filter()函数

filter()函数用于过滤序列。函数接受一个func与数列,函数一次作用于元素,根据布尔返回值决定是否返回元素。True则返回,False则不返回。

>>>list(filter(lambda x:x%2 == 1, [1,2,3,4,5]))
[1,3,5]
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 204,921评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,635评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,393评论 0 338
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,836评论 1 277
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,833评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,685评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,043评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,694评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 42,671评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,670评论 2 321
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,779评论 1 332
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,424评论 4 321
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,027评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,984评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,214评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,108评论 2 351
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,517评论 2 343

推荐阅读更多精彩内容