1.7.5 Example: Partitions
问题:用最大不超过m的正整数来划分n,有多少种方式?
比如,n = 9,m = 6时,划分方式为:
- 6 = 2 + 4
- 6 = 1 + 1 + 4
- 6 = 3 + 3
- 6 = 1 + 2 + 3
- 6 = 1 + 1 + 1 + 3
- 6 = 2 + 2 + 2
- 6 = 1 + 1 + 2 + 2
- 6 = 1 + 1 + 1 + 1 + 2
- 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