BZOJ3656: 異或

BZOJ3656

據說這個題是Cmn mod P 就做了。。。
一個位置p 寫成了pc wa了不知道多少遍。。

【代碼】

#include <cstdio>
#include <iostream>
#include <algorithm>
#define N 100005
#define INF 0x7fffffff
using namespace std;
typedef long long ll;

ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,Mod,cnt;
ll ans,tot,tmp; 
ll Fac[N],P[N],Pc[N];

void Divide(int x)
{
    for(int i=2;i*i<=x;i++) if(x%i==0)  
    {  
        P[++cnt]=i;Pc[cnt]=1;  
        while(x%i==0) Pc[cnt]*=i,x/=i;  
    }  
    if(x>1) P[++cnt]=Pc[cnt]=x;  
}

void Exgcd(ll a,ll b,ll &x,ll &y){
    if(b){
        Exgcd(b,a%b,y,x);
        y-=(a/b)*x;
    }
    else x=1,y=0;
}

ll Qpow(ll x,ll y,ll mod){
    ll rtn=1;
    while(y) {
        if(y&1) rtn=rtn*x%mod;
        x=x*x%mod;y>>=1;
    }
    return rtn;
}

ll Inv(ll a,ll b){
    ll x,y;
    Exgcd(a,b,x,y);
    return (x%b+b)%b;
}

ll Calc(ll n,ll p,ll pc) {
    if(n<p) return Fac[n];
    tot+=n/p;
    return Fac[n%pc]*Qpow(Fac[pc-1],n/pc,pc)%pc*Calc(n/p,p,pc)%pc;
}

ll Get_Fact(int p,int pc)
{
    Fac[0]=1;
    for(int i=1;i<pc;i++) Fac[i]=Fac[i-1]*(i%p?i:1)%pc;
    tot=0;
    ll t1=Calc(n,p,pc),t2=1;
    ll tmp=tot;
    for(int i=1;i<pc;i++) Fac[i]=Fac[i-1]*(i%p?Inv(i,pc):1)%pc;
    tot=0;
    t2=Calc(m,p,pc)*Calc(n-m,p,pc)%pc;
    return t1*t2%pc*Qpow(p,tmp-tot,pc)%pc;
}

int main()
{
    n=read(),m=read(),Mod=read();
    Divide(Mod);
    for(int i=1;i<=cnt;i++)
    {
        ll t1=Get_Fact(P[i],Pc[i]),Mi=Mod/Pc[i],t2=Inv(Mi,Pc[i]);
        ans=(ans+t1*Mi%Mod*t2%Mod)%Mod;
    }
    printf("%lld\n",ans);
    return 0;
}