Question

我最近问了this问题并得到了第一个答案。我想把它放到python代码中。这就是我所拥有的，但我一直得到0作为答案。

def f(n, k, s):
    ans = 0
    for j in range(1, min({k,s}) + 1):
        print j
        if (n == 1):
            if (k >= s):
                ans = ans + 1
            elif (k < s):
                ans = ans + 0
            elif (s > n):
                ans = ans + 0
        elif (n*k < s):
            ans = ans + 0
        else:
            ans = ans + f(n-1,j,s-j)
    return ans

print f(10, 12, 70)

我的代码出了什么问题？我需要改变什么？我不知道出了什么问题。请帮忙。谢谢！

Answer 1

您的代码太复杂了。您可以在数学交换中写出几乎一对一的答案：

def f(n, k, s):
    if n == 1:
        return int(k >= s)
        # or: 1 if k >=s else 0 
    return sum(f(n-1, j, s-j) for j in range(1, min(k, s)+1))
    # to make it faster:
    #return sum(f(n-1, j, s-j) for j in range(1, min(k, s)+1) if n*k >= s)

代码中的问题是你将基本案例检查放在循环中，当它应该在外面时：

def f(n, k, s):
    ans = 0
    if n == 1:
        return int(k >= s)

    for j in range(1, min({k,s}) + 1):
        print j
        if n*k >= s:
            ans += f(n-1,j,s-j)
    return ans

对于这两种实现，我得到了f(10, 12, 70)的结果为12660。

Answer 2

我不知道为什么你的工作不起作用，但这是一个实现，IMO更具可读性：

from itertools import permutations


def f(n, k, s):

    if k > s:
        k = s-1

    count = 0
    sum_perms = []

    number_list = []
    for i in range(1,k):
        for j in range(1,k,i):
            number_list.append(i)

    for perm in permutations(number_list, n):
        if sum(perm) == s and perm not in sum_perms:
            sum_perms.append(perm[:])
            count += 1

    return sum_perms, count

它比递归技术慢了很多： - （

itertools太棒了。

python中的递归公式用于递归sigma如何？

2 个答案: