Question

我正在研究一个发现所有k位数字的数字加起来等于给定n的问题。

我找到了解决方法，并将其作为整数分区问题进行了处理，但是我希望只能输入n和k个数字（没有max_element），但是当我尝试从代码中删除它时却没有似乎可以工作了。

我该如何改变它并逆转它？

def c(n, k, max_element):
    allowed = range(max_element, 0, -1)

    def helper(n, k, t):
        if k == 0:
            if n == 0:
                yield t
        elif k == 1:
            if n in allowed:
                yield t + (n,)
        elif 1 * k <= n <= max_element * k:
            for v in allowed:
                yield from helper(n - v, k - 1, t + (v,))
    return helper(n, k, ())

for p in c(5, 3, 3):
    print(p)

我尝试使用反向方法，但显然在生成器中不起作用。

结果：

(3, 1, 1)
(2, 2, 1)
(2, 1, 2)
(1, 3, 1)
(1, 2, 2)
(1, 1, 3)

预期结果：

113 122 131 212 221 311

Answer 1

这里有两个问题；首先是由于range(max_element, 0, -1)，您希望顺序编号，并且此代码以相反的顺序生成它们。另一个问题是，由于要生成数字，因此最小元素应为0，最大元素应始终为9。我们可以通过将范围更改为range(10)来解决这两个问题。

我们仍然需要注意不要生成以0开头的数字，因此我们将allowed作为参数并将range(1, 10)仅用于第一个数字。我还对其进行了更改，以将结果作为整数而不是元组返回。

作为参考，此生成器函数的代码来自my answer to another question。

def c(n, k):
    def helper(n, k, t, allowed):
        if k == 0:
            if n == 0:
                yield t
        elif k == 1:
            if n in allowed:
                yield 10*t + n
        elif 0 <= n <= 9 * k:
            for v in allowed:
                yield from helper(n - v, k - 1, 10*t + v, range(10))

    return helper(n, k, 0, range(1, 10))

示例：

>>> for p in c(5, 3):
...     print(p)
...
104
113
122
131
140
203
212
221
230
302
311
320
401
410
500

Answer 2

此功能可以解决问题

def c(n, k, max_element):
    allowed = range(max_element, 0, -1)

    def helper(n, k, t):
        if k == 0:
            if n == 0:
                yield t
        elif k == 1:
            if n in allowed:
                yield t + (n,)
        elif 1 * k <= n <= max_element * k:
            for v in allowed:
                yield from helper(n - v, k - 1, t + (v,))
    return helper(n, k, ())

def reversed_iterator(iter):
    return reversed(list(iter))

for p in reversed_iterator(c(5, 3, 3)):
    print(p)

这是输出：

(1, 1, 3)
(1, 2, 2)
(1, 3, 1)
(2, 1, 2)
(2, 2, 1)
(3, 1, 1)

递归生成所有数字总和为n的k位数字

2 个答案: