light oj 1067 Combinations (組合數的lucas定理)

/*
 用到了lucas定理：A、B是非負整數，p是質數。AB寫成p進制：A=a[n]a[n-1]...a[0]，B=b[n]b[n-1]...b[0]。
則組合數C(A,B)與C(a[n],b[n])*C(a[n-1],b[n-1])*...*C(a[0],b[0])  modp同
即：Lucas(n,m,p)=c(n%p,m%p)*Lucas(n/p,m/p,p) 


對於單獨的C(ni, mi) mod p，已知C(n, m) mod p = n!/(m!(n - m)!) mod p。顯然除法取模，這裏要用到m!(n-m)!的逆元。


根據費馬小定理：


已知(a, p) = 1，則 a^(p-1) ≡ 1 (mod p),  所以 a*a^(p-2) ≡ 1 (mod p)。


也就是 (m!(n-m)!)的逆元爲 (m!(n-m)!)^(p-2) ；
*/

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;
#define ll long long
#define maxn 1000005
#define MOD 1000003
ll fac[maxn];
ll pow_mod(ll a,ll b,ll mod){  //快速冪運算
    ll ret=1;
    while(b){
        if(b&1) ret=(ret*a)%mod;
        a=(a*a)%mod;
        b>>=1;
    }
    return ret;
}
ll get_fac(ll p){  //fac數組存的是前i個數的乘積
    fac[0]=1;
    for(int i=1;i<=p;i++) fac[i]=fac[i-1]*i%p;
}
ll lucas(ll n,ll m,ll p){ //lucas定理：Lucas(n,m,p)=c(n%p,m%p)*Lucas(n/p,m/p,p)
    ll ret=1;
    while(n&&m){
        ll a=n%p,b=m%p;
        if(a<b) return 0;
        ret=(ret*fac[a]*pow_mod(fac[b]*fac[a-b]%p,p-2,p))%p; //(m!(n-m)!)的逆元爲 (m!(n-m)!)^(p-2) ；
        n/=p;
        m/=p;
    }
    return ret;
}
int main(){
    int i,j,t;
    ll n,m,p;
    p=MOD;
    get_fac(p);
    cin>>t;
    for(i=1;i<=t;i++){
        scanf("%lld%lld",&n,&m);
        printf("Case %d: %lld\n",i,lucas(n,m,p));
    }
    return 0;
}