Composing Programs 1.7 Recursive Functions

1.7.5   Example: Partitions

问题：用最大不超过m的正整数来划分n，有多少种方式？

比如，n = 9，m = 6时，划分方式为：

1. 6 = 2 + 4
2. 6 = 1 + 1 + 4
3. 6 = 3 + 3
4. 6 = 1 + 2 + 3
5. 6 = 1 + 1 + 1 + 3
6. 6 = 2 + 2 + 2
7. 6 = 1 + 1 + 2 + 2
8. 6 = 1 + 1 + 1 + 1 + 2
9. 6 = 1 + 1 + 1 + 1 + 1 + 1

9种。我们定义函数count_partitions(n, m)。比如n = 6，m = 4，应该如何计算，参考下图。

 每次计算都可以分为两部分，一部分是用不超过m = 4来划分n，另一部分没有用到，也就是用不超过m = m - 1 = 3来划分n。由于函数count_partitions(n, m)，它的语义是：用用最大不超过m的正整数来划分n，所以对于用到了m的部分，可以用count_partitions(n - m, m)来表示，也就是用不超过m的数来继续划分剩下的部分，注意m可能会大于n - m，所以可以用count_partitions(n - m, min(n - m, m))来代替；而对于用m - 1的部分，可以count_partitions(n, m - 1)，表示用不超过m - 1 的正整数来划分n有多少种方式。这样我们就能写出我们的递归函数。

>>> def count_partitions(n, m):
        """Count the ways to partition n using parts up to m."""
        if n == 0:
            return 1
        elif n < 0:
            return 0
        elif m == 0:
            return 0
        else:
            return count_partitions(n-m, m) + count_partitions(n, m-1)
>>> count_partitions(6, 4)
9
>>> count_partitions(5, 5)
7
>>> count_partitions(10, 10)
42
>>> count_partitions(15, 15)
176
>>> count_partitions(20, 20)
627