组合数

求C(n,m)%p;

概念:
组合数我们用C(n,m)表示,它代表在n个数中取m个数的方案。(这个概念主要用于将问题抽象到组合数上)。
公式:
1、C(n,m)=C(n,n-m);
2、C(n,m)=C(n-1,m-1)+C(n-1,m);

3、C(0,n)+C(1,n)+C(2,n)+C(3,n)+…+C(n,n)=2 ^ n,这个公式被我们成为二次项定理

求法:
对于组合数的求法挺多的:Lucas定理、递推、逆元

递推:
这个其实就是在推杨辉三角,这个主要是用于n和m<=2000。对于这种问题我们就比较适合递推。
适用范围:
n<=2000(n是组合数的下标的值,即C(n,m)中的n)

代码:

#include<iostream>
#include<cstdio>
using namespace std;
int C[1000][1000],mod=1e9+9;
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<=n;i++)//边界一定要全 
    {
        C[i][1]=i%mod;//对于i个数中取一个数方案是i 
        C[i][i]=C[i][0]=1;//在i个数中取i个数只有一种方案 
    }
    for(int i=2;i<=n;i++)//直接开始递推 
        for(int j=2;j<i;j++)//边界不要计算 
            C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;//根据第二个公式,也就是杨辉三角的递推式 
    printf("%d",C[n][m]);//输出答案 
    return 0;
}

阶乘逆元:
这个方法就是根据组合数的定义公式去求,根据C(n,m)=n!/(m!*(n-m)!),所以我们要预处理出所有的阶乘以及逆元。

我们定义inv[i]表示i!的逆元,
我们可以知道inv[i+1]=(1/(i+1)!) ^ -1,(因为csdn比较难渲染数学公式,所以如果不方便看的话可以自己在纸上写),我们同乘i+1就变成了,inv[i+1]*(i+1)=(1/i!) ^ -1=inv[i],所以我们可以得到:inv[i+1] * (i+1)=inv[i],所以我们先求出n!的逆元,再倒推回来。
适用范围:
n<mod(n的意义上同,因为如果n>mod的话,其中有的阶乘就会因为是mod的倍数而没有逆元,这样就错了)
板子代码(费马小定理):

#include<iostream>
#include<cstdio>
using namespace std;
long long inv[1000100],jc[1000100],mod=1e9+9;
long long pow(long long a,long long b)//快速幂 
{
    long long ans=1;
    while(b)
    {
        if(b%2==1)
            ans=(ans*a)%mod;
        a=(a*a)%mod;
        b=b>>1;
    }
    return ans;
}
void pre()
{
    jc[1]=1;//阶乘边界 
    for(int i=2;i<=1000000;i++)
        jc[i]=(jc[i-1]*i)%mod;//求阶乘 
    inv[1000000]=pow(jc[1000000],mod-2);//用费马小定理求 
    for(int i=999999;i>=0;i--)
        inv[i]=(inv[i+1]*(i+1))%mod;//倒推求阶乘逆元 
    return ;
}
int main()
{
    pre();
    int T;
    scanf("%d",&T);
    while(T--)
    {
        long long n,m;
        scanf("%lld %lld",&n,&m);
        printf("%lld\n",(((jc[n]*inv[m])%mod)*inv[n-m])%mod);//求答案,记得mod,一直mod,乘完就mod! 
    }
}

板子代码(拓展欧几里得):

#include<iostream>
#include<cstdio>
using namespace std;
int mod=1e9+9,inv[1000100],jc[1000100],x,y;
void gcd(int a,int b)//拓展欧几里得 
{
    if(b==0)
    {
        x=1;
        y=0;
        return ;
    }
    gcd(b,a%b);
    int k=x;
    x=y;
    y=k-a/b*y;
    return ;
}
void pre()
{
    jc[1]=1;
    for(int i=2;i<=1000000;i++)//求阶乘 
        jc[i]=(jc[i-1]*i)%mod;
    gcd(jc[1000000],mod);//求最大阶乘逆元 
    inv[1000000]=(x+mod)%mod;//赋值 
    for(int i=999999;i>=0;i--)//倒推阶乘逆元 
        inv[i]=(inv[i+1]*(i+1))%mod;
    return ;
}
int C(int n,int m)
{
    return (((1LL*jc[n]*inv[m])%mod)*inv[n-m])%mod;//组合数 
}
int main()
{
    int T;
    scanf("%d",&T);
    pre();
    while(T--)
    {
        int n,m;
        scanf("%d %d",&n,&m);//读入 
        printf("%d\n",C(n,m));//输出答案 
    }
    return 0;
}

Lucas定理:
这个定理个人觉得还好,主要是针对mod很小,如果用阶乘的话就有可能会是mod的倍数,这样逆元求组合数就gg了。可是对于递推又被卡的情况下我们就只能用这个了。
适用范围:
在递推和阶乘逆元都挂的时候,就用这个。
公式:
Lucas(n,m,mod)(n和m是C(n,m)中的,mod就是取模对象)=C(n%mod,m%mod) * Lucas(n/mod,m/mod,mod)。

C(n,m) % p = (C(n % p, m % p) * Lucas( n / p, m / p )) % p。(递归出口为m==0,return 1)
代码:
这个代码有两种,因为我们可以看见,在公式中我们是有组合数的,只不过范围降了下来,所以我们主要是用Lucas定理把原本卡死递推和阶乘逆元给救回来。所以我们才有两种代码。当mod比较小时我们可以采用递推,而mod比较大时我们可以用阶乘逆元。
递推:

#include<iostream>
#include<cstdio>
using namespace std;
int C[2010][2010],mod;
void pre()//预处理 
{
    for(int i=1;i<=mod;i++)
    {
        C[i][1]=i%mod;
        C[i][i]=C[i][0]=1;
    }
    for(int i=2;i<=mod;i++)
        for(int j=2;j<i;j++)
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
    return ;
}
int Lucas(int n,int m)
{
    if(m==0)//边界 
        return 1;
    return (C[n%mod][m%mod]*Lucas(n/mod,m/mod))%mod;//Lucas定理的公式 
}
int main()
{
    int T;
    scanf("%d %d",&T,&mod);
    pre();
    while(T--)
    {
        int n,m;
        scanf("%d %d",&n,&m);
        printf("%d\n",Lucas(n,m));//Lucas定理 
    }
    return 0;
}

阶乘逆元:
对于这个虽然有了小优化,可还是有一个条件:mod必须是质数才可以,要不然题目保证mod以内的阶乘不会是mod的倍数。
费马小定理:

#include<iostream>
#include<cstdio>
using namespace std;
int jc[1000100],inv[1000100],mod;
int pow(long long a,long long b)//快速幂 
{
    long long ans=1;
    while(b)
    {
        if(b%2==1)
            ans=(ans*a)%mod;
        a=(a*a)%mod;
        b=b>>1;
    }
    return (int)ans;
}
void pre()//预处理阶乘逆元 
{
    jc[1]=1;
    for(int i=2;i<mod;i++)
        jc[i]=(jc[i-1]*i)%mod;
    inv[mod-1]=pow(jc[mod-1],mod-2);//费马小定理 
    for(int i=mod-2;i>=0;i--)
        inv[i]=(inv[i+1]*(i+1))%mod;
}
int C(int n,int m)//组合数公式 
{
    return (((jc[n]*inv[m])%mod)*inv[n-m])%mod;
}
int Lucas(int n,int m)
{
    if(m==0)//边界 
        return 1;
    return (C(n%mod,m%mod)*Lucas(n/mod,m/mod))%mod;//Lucas定理公式 
}
int main()
{
    int T;
    scanf("%d %d",&T,&mod);
    pre();
    while(T--)
    {
        int n,m;
        scanf("%d %d",&n,&m);
        printf("%d\n",Lucas(n,m));//输出 
    }
    return 0;
}

拓展欧几里得:

#include<iostream>
#include<cstdio>
using namespace std;
int mod,inv[1000100],jc[1000100],x,y;
void gcd(int a,int b)//拓展欧几里得 
{
    if(b==0)
    {
        x=1;
        y=0;
        return ;
    }
    gcd(b,a%b);
    int k=x;
    x=y;
    y=k-a/b*y;
    return ;
}
void pre()//预处理 
{
    jc[1]=1;
    for(int i=2;i<mod;i++)
        jc[i]=(jc[i-1]*i)%mod;
    gcd(jc[mod-1],mod);//处理出解 
    inv[mod-1]=(x+mod)%mod;//把解取正 
    for(int i=mod-2;i>=0;i--)
        inv[i]=(inv[i+1]*(i+1))%mod;
    return ;
}
int C(int n,int m)//组合数尝龟 
{
    return (((1LL*jc[n]*inv[m])%mod)*inv[n-m])%mod;
}
int Lucas(int n,int m)
{
    if(m==0)
        return 1;
    return (1LL*C(n%mod,m%mod)*Lucas(n/mod,m/mod))%mod;
}
int main()
{
    int T;
    scanf("%d %d",&T,&mod);//多组测试数据,mod的值 
    pre();
    while(T--)
    {
        int n,m;
        scanf("%d %d",&n,&m);
        printf("%d\n",Lucas(n,m));//输出 
    }
    return 0;
}

 

例题:

 

 代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int mod =1e9+7;
ll quick(ll x,ll n) {
    ll res = 1;
    x%=mod;
    while (n) {
        if(n&1) res = res*x%mod;
        x = x*x %mod;
        n >>=1;
    }
    return res;
}
 
ll C(ll n,ll m) {
    if(n < m) return 0;
    // c(n,m) = n!/m!/(n-m)!
    ll res = 1;
    for(int i=1; i<=m; i++) {
        ll a = (n+i-m)%mod;
        ll b = i%mod;
        res = res*(a*quick(b,mod-2)%mod)%mod;
    }
    return res;
}
 
ll Lucas(ll n,ll m) {
    if(m == 0) return 1;
    return C(n%mod, m%mod) * Lucas(n/mod,m/mod)%mod;
}
int main(){
    ll m,n;
    string p;
    cin>>n>>m>>p;  //m<n
    ll ans = 0;
    for(int i=0;p[i]!='\0';i++){
        ans= (ans*10+p[i]-'0')%mod;
    }
    if(Lucas(n,m)==ans)
        cout<<"Yes!"<<endl;
    else
        cout<<"No!"<<endl;
    return 0;
}
posted @ 2020-02-24 21:24  sqsq  阅读(792)  评论(0编辑  收藏  举报