Python求解排列组合：递归程序设计艺术(4)

从5个不同的小球里任取3个的组合有多少？这是一个典型的组合问题，答案是C(5, 2)即10种。但是我们现在并不是要用数学方法求组合的解，而是要求编写一个递归函数comb(m, n)以求得从m个小球里任取n个的组合数，其中m ≥ n始终成立。

我们根据递归程序三步曲来考虑这个函数。

第一，确定递归边界。在什么情况下，组合数可以直接求得，而不必进行递归？显然当n=0时，不论m等于多少，组合数都是1。把n=0作为递归边界已经足够了。但是，如果还有其他边界则也应该考虑在内，这样有助于程序在递归过程中更快地接近边界从而提高程序运行速度，减少内存占用。如果n==m，则意味着把所有的小球都选中，这样的组合数也只有1个。

第二，递归假设。我们假设只要把m和/或n减1，则组合数总能求得出来。

第三，递归推导。我们把最后一个小球Z拎出来单独考虑。如果最后取出来的n个小球里不包含Z，这种情况相当于从除了Z之外的m-1个小球里取n个，根据递归假设，共有comb(m-1, n)种组合。反之，如果取出来的n个小球里包含Z，这种情况相当于从除了Z之外的m-1个小球里取n-1个小球的组合，根据递归假设，共有comb (m-1, n-1)种组合。代码如下：

解决组合问题的递归程序

def comb(m, n):
    if n == 0 or m == n:
        return 1
    return comb(m-1, n-1) + comb(m-1, n)

if __name__ == '__main__':
    for m in range(1, 10+1):
        for n in range(0, m+1):
            print('C(%d, %d) = %d' % (m, n, comb(m, n)))

上述代码只是计算了组合数，如果要计算所有可能的组合则只需把comb的返回值从整数改为列表，comb(m-1, n-1) + comb(m-1, n)改为两个列表的合并即可：

输出所有组合

def comb(m, n):
    if n == 0:
        return [[]]  # 仅有一个空组合
    if m == n:
        # 输出仅含有0～m-1的组合
        return [[e for e in range(m)]]
    # 计算含有最后一个小球(m-1)的所有组合
    result = [c + [m-1] for c in comb(m-1, n-1)]
    return result + comb(m-1, n)

if __name__ == '__main__':
    for m in range(1, 10+1):
        for n in range(0, m+1):
            print('C(%d, %d) = %s' % (m, n, comb(m, n)))

输出结果：

C(1, 0) = [[]]
C(1, 1) = [[0]]
C(2, 0) = [[]]
C(2, 1) = [[1], [0]]
C(2, 2) = [[0, 1]]
C(3, 0) = [[]]
C(3, 1) = [[2], [1], [0]]
C(3, 2) = [[1, 2], [0, 2], [0, 1]]
C(3, 3) = [[0, 1, 2]]
......

解决了组合数问题，请读者思考：如何解决排列问题呢？同样可以把最后一个小球Z拿出来单独考虑，如果最后取出来的n个小球里不包含Z，这种情况相当于从除了Z之外的m-1个小球里取n个的排列，根据递归假设，共有perm(m-1, n)种排列。反之，如果取出来的n个小球里包含Z，这种情况相当于从除了Z之外的m-1个小球里取n-1个小球的排列，然后把Z插入这n-1个小球之间，共有n个可能性，两个数相乘共有n * perm(m-1, n-1)种排列。代码如下：

解决排列问题的递归程序

def perm(m, n):
    if n == 0:
        return 1
    result = 0 if m == n else perm(m-1, n)  # 不包含第一个小球
    result += perm(m-1, n-1) * n;  # 包含第一个小球
    return result

if __name__ == '__main__':
    for m in range(1, 10+1):
        for n in range(0, m+1):
            print('C(%d, %d) = %d' % (m, n, perm(m, n)))

用类似的办法也可以输出所有的排列数，代码略。