[来源未知]组合数取模

组合数取模
【问题描述】
给定n,m,p，求C(n,m) mod p。
【输入说明】
第一行三个整数n,m,p。
【输出格式】
输出一个整数，表示C(n,m) mod p。
【样例输入】
5 2 3
【样例输出】
1
【数据规模及约定】
对于100%的数据，1≤m≤n≤10^6,1≤p≤10^5。

sol：
发现p很小，对p质因数分解，为p1^c1,p2^c2…..pn^cn,发现最多分成8个质因数。
然后我们对于一个数，也拆分成p1^d1*p2^d2*p3^d3*d（d为常数）的形式，那么d就和p互质了，用欧拉定理可以求逆元。

欧拉定理内容：
如果a和b互质
a^(phi(b)-1)为a在模b下的逆元。

#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
typedef long long ll;
int n,m,mod;
inline int read()
{
    char c;
    int res,flag=0;
    while((c=getchar())>'9'||c<'0') if(c=='-')flag=1;
    res=c-'0';
    while((c=getchar())>='0'&&c<='9') res=(res<<3)+(res<<1)+c-'0';
    return flag?-res:res;
}
int prime[1100000];
int ksm(int s,int t)
{
    int res=1;
    while(t)
    {
        if(t&1) res=(ll)res*s%mod;
        s=(ll)s*s%mod;
        t>>=1;
    }
    return res;
}
int phi[1100000];
int inv(int x)
{
    return ksm(x,phi[mod]-1);
}
struct cc
{
    int c[10];
    friend inline cc operator *(const cc &a,const cc &b)
    {
        cc c;
        memset(c.c,0,sizeof(c.c));
        for(int i=1;i<=prime[0];++i)
        c.c[i]=a.c[i]+b.c[i];
        c.c[0]=(ll)a.c[0]*b.c[0]%mod;
        return c;
    }
    friend inline cc operator /(const cc &a,const cc &b)
    {
        cc c;
        memset(c.c,0,sizeof(c.c));
        for(int i=1;i<=prime[0];++i)
        c.c[i]=a.c[i]-b.c[i];
        c.c[0]=(ll)a.c[0]*inv(b.c[0])%mod;
        return c;
    }
};
cc trans(int x)
{
    cc t;
    memset(t.c,0,sizeof(t.c));
    for(int i=1;i<=prime[0]&&x!=1&&prime[i]<=x;++i)
    if(x%prime[i]==0)
    {
        while(x%prime[i]==0)
        {
            x/=prime[i];
            t.c[i]++;
        }
    }
    t.c[0]=x;
    return t;
}
bool is[1100000];
int ans;
int main()
{
    freopen("com.in","r",stdin);
    freopen("com.out","w",stdout);
    n=read();
    m=read();
    mod=read();
    int e=sqrt(mod),tmp=mod;
    phi[1]=1;
    for(int i=2;i<=mod;++i)
    {
        if(!is[i])
        {
            prime[++prime[0]]=i;
            phi[i]=i-1;
        }
        for(int j=1;j<=prime[0]&&i*prime[j]<=mod;++j)
        {
            is[i*prime[j]]=1;
            if(!(i%prime[j]))
            {
                phi[i*prime[j]]=phi[i]*prime[j];
                break;
            }
            phi[i*prime[j]]=phi[i]*(prime[j]-1);
        }
    }
    prime[0]=0;
    for(int i=2;i<=e;++i)
    if(tmp%i==0)
    {
        prime[++prime[0]]=i;
        while(tmp%i==0) tmp/=i;
    }
    if(tmp!=1) prime[++prime[0]]=tmp;
    cc y,b,c;
    memset(b.c,0,sizeof(b.c));
    b.c[0]=1;
    memset(c.c,0,sizeof(c.c));
    c.c[0]=1;
    memset(y.c,0,sizeof(y.c));
    y.c[0]=1;
    for(int i=2;i<=n;++i)
    {
        y=y*trans(i);
        if(i==m) b=y;
        if(i==n-m) c=y;
    }
    cc t;
    t=y/b/c;
    ans=1;
    for(int i=1;i<=prime[0];++i)
        ans=(ll)ans*ksm(prime[i],t.c[i])%mod;
    printf("%d",(ll)ans*t.c[0]%mod);
}