python求组合数c_Python快速求组合数C(n,m)三种方法整理

百度百科对于组合数的定义是：从n个不同元素中，任取m(m≤n)个元素并成一组，叫做从n个不同元素中取出m个元素的一个组合；从n个不同元素中取出m(m≤n)个元素的所有组合的个数，叫做从n个不同元素中取出m个元素的组合数。

由于经常遇到一些组合数问题，所以整理一些常见的快速求组合数的方法，附上Python的实现代码。

一、m,n不是特别大的时候：

(nm)=n!m!∗(n−m)!(nm)=n!m!∗(n−m)!\binom {n}{m}=\frac {n!}{m!*(n-m)!}(mn​)=m!∗(n−m)!n!​

可以直接调用math.factorial求得阶乘，然后算出组合数，如下：

import math

n,m = map(int,input().split())

print(math.factorial(n)//(math.factorial(m)*math.factorial(n-m)))

输入：

5 3

输出：

10

二、用定义式递归：

(nm)=(n−1m−1)+(n−1m)(nm)=(n−1m−1)+(n−1m)\binom {n}{m}=\binom {n-1}{m-1}+\binom {n-1}{m}(mn​)=(m−1n−1​)+(mn−1​)

递归出口就在于当n=m或者m=1的时候。

n,m = map(int,input().split())

def rec(n,m):

if m == n:

return 1

elif m == 1:

return n

else:

return rec(n-1,m-1)+rec(n-1,m)

print(rec(n,m))

输入：

10 3

输出：

120

三、逆元+快速幂思想参考大佬

前面的两种方法，在n,m数字很大的时候，运行时间会很长。在介绍第三个方法之前，先来介绍几个概念，不当之处，欢迎指点。

（1）、同余定理

百度百科：同余定理数论中的重要概念。给定一个正整数m，如果两个整数a和b满足a-b能够被m整除，即(a-b)/m得到一个整数，那么就称整数a与b对模m同余，记作a≡b(mod m)。对模m同余是整数的一个等价关系。

5 ≡ 3(mod 2) #5和 3对模2同余

（2）、模的加减乘除运算

取模运算的等价变形适合加法、减法、乘法

(a+b)%p=(a%p+b%p)%p(a+b)%p=(a%p+b%p)%p(a + b) \% p = (a \% p + b \% p) \% p(a+b)%p=(a%p+b%p)%p

(a−b)%p=(a%p−b%p)%p(a−b)%p=(a%p−b%p)%p(a - b)\% p = (a\% p - b\% p) \% p(a−b)%p=(a%p−b%p)%p

(a∗b)%p=(a%p∗b%p)%p(a∗b)%p=(a%p∗b%p)%p( a * b) \% p = (a \% p * b \% p) \% p(a∗b)%p=(a%p∗b%p)%p

但是，取模运算的等价变形不符合除法

a/b%p≠(a%p/b%p)%pa/b%p≠(a%p/b%p)%pa/b \% p ≠ (a\%p / b\%p) \% pa/b%p​=(a%p/b%p)%p

比如：

（100%20）%11=5%11=5（100%20）%11=5%11=5（100 \% 20）\% 11 = 5 \% 11 = 5（100%20）%11=5%11=5

（100%11/20%11)%11=(1/9)%11=0%11=0（100%11/20%11)%11=(1/9)%11=0%11=0（100\%11 / 20\%11) \% 11 = (1 / 9) \% 11 = 0 \% 11 = 0（100%11/20%11)%11=(1/9)%11=0%11=0

（3）、逆元

逆元：对于a和p，若a和p互素且

(a∗b)%p≡1(a∗b)%p≡1(a * b) \% p ≡ 1(a∗b)%p≡1

则称b为a%p的逆元。

假设c为b%p的逆元，即b∗c%p=1b∗c%p=1b*c\%p=1b∗c%p=1

(a/b)%p(a/b)%p(a / b) \% p(a/b)%p

=(a/b)∗1%p=(a/b)∗1%p= (a / b) * 1 \% p=(a/b)∗1%p

=(a/b)∗(b∗c%p)%p=(a/b)∗(b∗c%p)%p=(a/b)*(b*c\%p)\%p=(a/b)∗(b∗c%p)%p

=a∗c%p=a∗c%p=a*c\%p=a∗c%p

=(a%p)∗(c%p)%p=(a%p)∗(c%p)%p=(a\%p)*(c\%p)\%p=(a%p)∗(c%p)%p

这样就把求(a/b)%p(a/b)%p(a / b) \% p(a/b)%p转化成一个(a%p)∗(c%p)%p(a%p)∗(c%p)%p(a\%p)*(c\%p)\%p(a%p)∗(c%p)%p乘法问题。

（4）、费马小定理

百度百科：费马小定理(Fermat’s little theorem)是数论中的一个重要定理，在1636年提出。如果p是一个质数，而整数a不是p的倍数，则有a(p−1)%p≡1a(p−1)%p≡1a ^ {(p-1)} \% p ≡ 1a(p−1)%p≡1。

a（p−1）=a（p−2）∗aa（p−1）=a（p−2）∗aa^{（p-1）}=a^{（p-2）} * aa（p−1）=a（p−2）∗a

所以有a（p−2）∗a%p≡1a（p−2）∗a%p≡1a^{（p-2）} * a\%p≡1a（p−2）∗a%p≡1

所以a（p−2）%pa（p−2）%pa^{（p-2）}\%pa（p−2）%p是aaaa的逆元。

由于m,n很大，现在要求的是(nm)%p(nm)%p\binom{n}{m}\%p(mn​)%p，假设pppp取100000007，由于取模运算的等价变形不适用于除法，即：

(nm)%p≠n!%pm!%p∗(n−m)!%p%p(nm)%p≠n!%pm!%p∗(n−m)!%p%p\binom{n}{m}\%p≠ \frac{n!\%p}{m!\%p*(n-m)!\%p}\%p(mn​)%p​=m!%p∗(n−m)!%pn!%p​%p

根据上面求得：

(a/b)%p=(a%p)∗(c%p)%p(a/b)%p=(a%p)∗(c%p)%p(a / b) \% p=(a\%p)*(c\%p)\%p(a/b)%p=(a%p)∗(c%p)%p

c是b%p的逆元。c是b%p的逆元。c是b\%p的逆元。c是b%p的逆元。

就相当于我们要求出m!和(n−m)!m!和(n−m)!{m!和(n-m)!}m!和(n−m)!的逆元，根据a（p−2）%pa（p−2）%pa^{（p-2）}\%pa（p−2）%p是aaaa的逆元，得到

m!的逆元是m!(p−2)m!的逆元是m!(p−2)m!的逆元是m!^{(p-2)}m!的逆元是m!(p−2)

(n−m)!的逆元是(n−m)!(p−2)(n−m)!的逆元是(n−m)!(p−2)(n-m)!的逆元是(n-m)!^{(p-2)}(n−m)!的逆元是(n−m)!(p−2)

所以(nm)%p=[n!%p∗(m!(p−2)%p)∗((n−m)!(p−2)%p)]%p(nm)%p=[n!%p∗(m!(p−2)%p)∗((n−m)!(p−2)%p)]%p\binom{n}{m}\%p=[n!\%p*(m!^{(p-2)}\%p)*((n-m)!^{(p-2)}\%p)]\%p(mn​)%p=[n!%p∗(m!(p−2)%p)∗((n−m)!(p−2)%p)]%p

下面用代码来实现：

import math

n,m = map(int,input().split())

p = 100000007

def power(x,y): #求x的y次方

p = 100000007

res = 1

while y:

if y % 2 != 0:

res *= (x%p)

y >>= 1

x *= (x%p)

return res

a = (math.factorial(n))%p

b = (power(math.factorial(m),(p-2)))%p

c = (power(math.factorial(n-m),(p-2)))%p

print(a*b*c%p)

中间的power函数是用快速幂做的，直接用Python的运算符“**”的话会非常大，会超时。

小白欢迎大佬指点~