打印可以等于给定数字的所有 3 个连续数字

Question

我如何编写一个递归回溯函数count(N,S)，它打印所有N-digit 数字，使得数字中每3个连续数字的总和完全等于S，其中N将是小于等于 10，取值范围为 0 到 27。

代码：

def count(S):
    n = int(S)
    if n % 3 == 0:
        print(int(n / 3 - 1),int(n / 3),int(n / 3 + 1))
    else:
        print(None)
S = 27
count(S)

样本输出：

8 9 10

我很困惑如何递归编写。

Answer 1

您当前的函数不是递归的。为了使其递归，您基本上必须在count(n, s) 的执行中的某个地方调用count(n-1, s)。一种方法是这样的：

• 如果n > 1，获取n-1 的可能解决方案并附加任何仍然满足条件的数字
• 如果n == 0 只返回""（如果函数返回字符串而不是实际整数会更容易）

作为一个生成器函数，这可能看起来有点像这样。当然，您也可以将结果收集到一个列表中并return 他们，或者只是获取这些数字的计数并将其返回。

def count(n, s):
    if n > 0:
        for x in count(n-1, s):
            for d in range(10):
                y = str(d) + x
                if len(y) < 3 or sum(map(int, y[:3])) == s:
                    yield y
    else:
        yield ""

for x in count(5, 15):
    print(x)