组合数求解总结

组合数

意图计算$C_a^b$的值，此处列举三种常用方法。

简单组合计数

$a,b$很小（小于等于3000左右），采用递推的方法，根据公式$C_a^b=C_{a-1}^{b-1}+C_{a-1}^b$,用二维数组$C[i][j]$表示$C_i^j$。

#include<iostream>
#define N 2010
using namespace std;
typedef long long ll;
ll C[N][N];
void init(){//预处理打表2000以内的组合数。
	for(int i = 0;i<=2000;++i){
		for(int j = 0;j<=i;++j){
			if(!j) C[i][j] = 1;
			else C[i][j] = C[i-1][j]+C[i-1][j-1];
		}
	}
}
int main(){
	int n,a,b;
	init();//打表
	cin >> n;
	while(n--){
		cin >> a >> b;
		cout << C[a][b]<<"\n";//查询
	}
	return 0;
}	

阶乘逆元求解

$a,b\le 100000$时，已经不能采用简单组合计数的方法，时间复杂度过高。这里可以利用公式$C_a^b=\frac{a!}{(a-b)!\times b!}$,将1~n的阶乘以及其逆元计算得出即可。另外，应用这类方法时，通常都会进行取模运算，此处以mod = 1e9+7为例。
代码如下：

#include<iostream>
using namespace std;
typedef long long ll;
const ll mod = 1e9+7;
ll fact[100010],infact[100010];
ll qpow(ll a,ll p){
	ll ans = 1;
	while(p){
		if(p&1) ans = (ans*a)%mod;
		p>>=1;
		a*=a;
		a%=mod;
	}
	return ans%mod;
}
int main(){
	int n,a,b;
	cin >>n;
	fact[0] = infact[0] = 1;
	for(int i = 1;i<=100000;++i){
		fact[i] = fact[i-1]*i%mod;
		infact[i] = infact[i-1]*qpow(i,mod-2)%mod;
		//逆元i乘以i-1的阶乘逆元，结果是i的阶乘逆元
		//也可以用i的阶乘逆元乘以i,反推得到i-1的阶乘逆元
	}
	while(n--){
		cin >> a >> b;
		//三项直接相乘可能会long long 溢出
		ll ans = (fact[a]*infact[a-b]%mod)*infact[b]%mod;
		cout << ans <<"\n";
	}
	return 0;
}

Lucas定理

$1\le a,b\le 10^{18},模数1\le p\le 10^5$时打表已经不可取了，此处应采用卢卡斯定理$C_a^b\equiv C_{a\%p}^{b\%p}\times C_{a/p}^{b/p}(\mathrm{m}\mathrm{o}\mathrm{d}p)$，当a,b都小于p，而p的大小在$10^5$,可以通过定义递推得到$C_a^b$,若有比p大，则继续调用Lucas定理求解。

#include<iostream>
using namespace std;
typedef long long ll;
ll fact[100010],infact[100010];
ll mod;
ll qpow(ll a,ll p){
	ll ans = 1;
	while(p){
		if(p&1) ans = (ans*a)%mod;
		p>>=1;
		a*=a;
		a%=mod;
	}
	return ans%mod;
}
ll c(int a,int b){//a,b<=mod，规模小
	ll res = 1;
	for(int i = 1,j=a;i<=b;++i,--j){
		res = res * j % mod;
		res = res * qpow(i,mod-2) % mod;
	}
	return res;
}
ll Lucas(ll a,ll b){//卢卡斯定理
	if(a<mod&&b<mod) return c(a,b);
	return c(a%mod.b%mod)*Lucas(a/mod,b/mod)%mod;
}
int main(){
	ll a,b,n;
	cin >>n;
	while(n--){
		cin >> a >> b >> mod;
		cout << Lucas(a,b);
	}
	return 0;
}