Python:递归生成器

皮里克·马佐迪耶

我想递归地生成一串数字的切片的所有可能的总和。例如:

Input: '891'
Output: 18, 99, 90, 891

In details:
    8+9+1 = 18
    8+91 = 99
    89+1 = 90
    891 = 891

但是,我编写的代码产生了generators的generators:

# s: string of digits, c: sum of slices
def rec(s,c):
    for i in range(1,len(s)+1):
        if s=='': yield c
        else: yield rec(s[i:],c+int(s[:i]))

我该如何解决?

奥尔普

有时候正确可能会很棘手。但是您通过添加额外的参数使自己感到困惑:

def rec(s):
    # No split.
    yield int(s)

    # Split.
    for i in range(1, len(s)):
        left = int(s[:i])
        for right in rec(s[i:]):
           yield left + right

确实:

>>> list(rec("891"))
[891, 99, 18, 90]

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章