525RPA.CoM

广告

RPA开发教程-【算法题】不定列表求和

182人参与 |分类: 产品百科|时间: 2019年11月01日

问题

在一个嵌套了多个元素为数字或列表 (列表中也是数字或列表) 的列表中,求出所有数字的和。
例如:[1, [2, [3, [4, 5]]]]的和为 15,[1, 2, 3, 4, 5]的和也为 15。

分析

通过遍历每一个元素,如果发现这个元素不是列表,就把它累加,否则继续遍历这个元素直到找到数字。可以采用递归的方式来求解。

def sum_ulist(exp):
    sum = 0
    for i in exp:
        sum += i if type(i) != list else sum_ulist(i)    return sum

可以看到最后的程序非常短小,运行如下:

>>> sum_ulist([1, [2, [3, [4, 5]]]])15>>> sum_ulist([1, 2, 3, 4, 5])15

思考

还是一个不定嵌套层数的列表,这次不进行求和,而是把其中的所有数字都取出来,最后放在一个列表中返回。


与上面类似,不过上面的求和在每一次函数调用中只关注当前层级的总和即可,而这边的递归需要把数字都放在同一个列表中,可以利用默认参数的一个特性来解决。

我们之前提到,默认参数必须设置为不可变,否则会带来一些问题:当你调用此函数却不指定默认参数的值时,它会自动使用函数第一次调用中默认参数的对象,如果这个对象发生了改变,可能达不到我们预期的效果。

举个例子:

def a(li=[]):
    li.append(1)    return li>>> a()
[1]>>> a()
[1, 1]>>> a()
[1, 1, 1]

这种情况在很多时候都不是我们想要的,这边却可以利用这个特性:

def add_ulist(exp, tem_li=[]):    for i in exp:
        tem_li.append(i) if type(i) != list else add_ulist(i)
    return sum

通过在 else 语句后面的递归中不指定默认参数,我们就让后面的函数调用都只操作第一次函数调用时初始化的 tem_li 对象了,少写了几个字母 :)

运行如下:

>>> add_ulist([1, [2, [3, [4, 5]]], [6, [7, 8]]])[1, 2, 3, 4, 5, 6, 7, 8]

当然,在 else 语句后面的递归中将这次函数获得的 tem_li 列表传给默认参数也可以:

def add_ulist(exp, tem_li=[]):    for i in exp:
        tem_li.append(i) if type(i) != list else add_ulist(i, tem_li)
    return sum

最后,如果弄不清楚默认参数的奇妙性质的话,还是老老实实将默认参数设置为不可变对象吧:

def add_ulist(exp, tem_li=None):
    if tem_li is None:
        tem_li = []
    for i in exp:
        tem_li.append(i) if type(i) != list else add_ulist(i, tem_li)
    return sum



来源:[jane小姐](微信/QQ号:1055692322),转载请保留出处和链接!

地址:

必填

选填

选填

◎已有 0 人评论,请发表您的观点。

  
    
这个是评论广告
广告