ACM数学之求组合数

递推法

• 根据公式，$C_a^b=C_{a-1}^b+C_{a-1}^{b-1}$
• 从实际意义理解：在a个苹果里面选b个苹果的方案：，先假设有一个苹果x，如果我们选了x，那么就在剩下的a-1个苹果里面选b-1个，如果不选x,那么就是在a-1个苹果里选b个
• 时间复杂度：$O(n^2)$
• 例题：求组合数1

AC代码

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof a)
using namespace std; 
typedef long long ll;
const int N = 2e3 + 10, mod = 1e9 + 7;
int c[N][N];
void init()
{
    //c[0][0]=1;
	for(int i=0; i<N; 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])%mod;     
		
		
		
}
int main()
{
	init();
    int t;
    cin>>t;
    while(t--)
    {
    	int a,b;
    	scanf("%d%d",&a,&b);
    	printf("%d\n",c[a][b]);
    	
	}
    return 0;
}

通过预处理逆元的方式求组合数

• 就是 根据定义公式：$C_a^b=\frac{a!}{(a-b)!\ast b!}$,
  我们预处理出阶层f数组和逆元数组inf[i]，f[i]=f[i-1]*i%mod，
  inf[i]=inf[i-1]*qmi(i,mod-2)

• 时间复杂度$O(n)$

• 例题：组合数2

AC代码

#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof a)
#define ll  long long
using namespace std; 
const int N = 1e5+9 + 10, mod = 1e9 + 7;
ll f[N],inf[N];
int qmi(int x,int k)
{
	int res=1;
	while(k)
	{
		if(k&1) res=1ll*res*x%mod;
		k>>=1;
		x=1ll*x*x%mod;
	}
	return res;
}
void init()
{
   	f[0]=inf[0]=1;
	for(int i=1; i<N; i++)
	{
		f[i]=1ll*f[i-1]*i%mod;
		inf[i]=1ll*inf[i-1]*qmi(i,mod-2)%mod;
	}  		
}
int main()
{
	init();
    int t;
    cin>>t;
    while(t--)
    {
    	int a,b;
    	scanf("%d%d",&a,&b);
    	printf("%lld\n",1ll*f[a]*inf[b]%mod*inf[a-b]%mod);
	}
    return 0;
}

卢卡斯定理求组合数

• 这里觉得以为大佬写的很不错，就直接放链接了
  大佬博客

• 时间复杂度：$O(lo{g}_p^n\ast p)$, 适用于a,b很大，但是mod(mod<=1e5)很小时求组合数,

AC代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e5 + 10;
int mod;
int f[N],inf[N]; 
int qmi(int x,int k)
{
	int res=1;
	while(k)
	{
		if(k&1) res=1ll*res*x%mod;
		k>>=1;
		x=1ll*x*x%mod; 
	 } 
	 return res;
}
int C(int a,int b)
{
	int res=1;
	for(int i=1, j=a; i<=b; i++,j--)
	{
		res=1ll*res*j%mod;
		res=1ll*res*qmi(i,mod-2)%mod;
	}
	return res;
}
int lucas(ll a,ll b)
{
	if(a<mod && b<mod) return C(a,b);
	else return 1ll*C(a%mod,b%mod)*lucas(a/mod,b/mod)%mod;
} 
int main()
{
     
    
     int t;
     cin>>t;
     while(t--)
     {
     	 ll a,b;
     	 cin>>a>>b>>mod;
    
     	 cout<<lucas(a,b)<<endl;
	 }
    return 0;
}

总结

这里记录下自己的学习历程，求组合数时要根据不同的场景选择适当的方式。欢迎大家留言交流