一.一般粗暴法
用普通的组合公式来算
当求C(n,m)%MOD时容易超范围,因为这种方法计算有除法,不能边做边MOD.
优点:复杂度O(n),缺点只能求n,m极小的情况。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll MOD = 1e9 + 7;
ll cal(int n,int m)
{
ll ans=1;
for(int i=n;i>=n-m+1;i--)
{
ans*=(ll)i;
}
for(int i=1;i<=m;i++)
{
ans/=(ll)i;
}
return ans%MOD;
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
printf("%lld\n",cal(n,m));
return 0;
}
二.杨辉三角法
首先介绍一下杨辉三角法公式
C(n,m)=C(n-1,m)+C(n-1,m-1)
复杂度O(n^2),但由于是加法,可以边做边MOD,保证数字不会超限。
#include<cstdio>
const int N = 2000 + 5;
const int MOD = (int)1e9 + 7;
long long comb[N][N];//comb[n][m]就是C(n,m)
void init()
{
for(int i = 0; i < N; i ++)
{
comb[i][0] = comb[i][i] = 1;
for(int j = 1; j < i; j ++)
{
comb[i][j] = comb[i-1][j] + comb[i-1][j-1];
comb[i][j] %= MOD;
}
}
}
int main()
{
init();
int n,m;
while(~scanf("%d%d",&n,&m))
printf("%d\n",comb[n][m]);
}
三.LUCAS定律
LUCAS定律:C(n,m)%p=C(n/p,m/p)*C(n%p,m%p)%p
证明:
最后一步的得出详见https://blog.csdn.net/qq_40679299/article/details/80489761(其实不难,自己也可以一眼看出来)
总条件:p为质数,p为非质数时可以用扩展卢卡斯定律,但这里不讨论!
模板一.适用于n,m,p较大的情况,复杂度大概为O(mlogn)
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll p = (ll)1e9 + 7;
ll pow(ll a, ll b, ll m)
{
ll ans = 1;
a %= m;
while(b)
{
if(b & 1)ans = (ans % m) * (a % m) % m;
b /= 2;
a = (a % m) * (a % m) % m;
}
ans %= m;
return ans;
}
ll inv(ll x, ll p)//x关于p的逆元,p为素数
{
return pow(x, p - 2, p);
}
ll C(ll n, ll m, ll p)//组合数C(n, m) % p
{
if(m > n)return 0;
ll up = 1, down = 1;//分子分母;
for(int i = n - m + 1; i <= n; i++)up = up * i % p;
for(int i = 1; i <= m; i++)down = down * i % p;
return up * inv(down, p) % p;
}
ll Lucas(ll n, ll m, ll p)
{
if(m == 0)return 1;
return C(n % p, m % p, p) * Lucas(n / p, m / p, p) % p;
}
int main()
{
long long n,m;
while(~scanf("%lld%lld",&n,&m))
{
printf("%lld\n",Lucas(n,m,p));
}
return 0;
}
模板二,打表版,适用于n,m较大,p<10^6情况。复杂度为O(logn)
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 1e5 + 10;
const ll p = (ll)13;
ll fac[maxn];//阶乘打表
void init(ll p)//此处的p应该小于1e5,这样Lucas定理才适用
{
fac[0] = 1;
for(int i = 1; i <= p; i++)
fac[i] = fac[i - 1] * i % p;
}
ll pow(ll a, ll b, ll m)
{
ll ans = 1;
a %= m;
while(b)
{
if(b & 1)ans = (ans % m) * (a % m) % m;
b /= 2;
a = (a % m) * (a % m) % m;
}
ans %= m;
return ans;
}
ll inv(ll x, ll p)//x关于p的逆元,p为素数
{
return pow(x, p - 2, p);
}
ll C(ll n, ll m, ll p)//组合数C(n, m) % p
{
if(m > n)return 0;
return fac[n] * inv(fac[m] * fac[n - m], p) % p;
}
ll Lucas(ll n, ll m, ll p)
{
if(m == 0)return 1;
return C(n % p, m % p, p) * Lucas(n / p, m / p, p) % p;
}
int main()
{
ll n,m;
init(p);
while(~scanf("%lld%lld",&n,&m))
{
printf("%lld\n",Lucas(n,m,p));
}
return 0;
}