luogu3807 【模板】卢卡斯定理

http://www.elijahqi.win/archives/3174
题目背景

这是一道模板题。

题目描述

给定n,m,p(

1\le n,m,p\le 10^5

1≤n,m,p≤105 )

求

C_{n+m}^{m}\ mod\ p

Cn+mm​ mod p

保证P为prime

C表示组合数。

一个测试点内包含多组数据。

输入输出格式

输入格式：

第一行一个整数T(

T\le 10

T≤10 )，表示数据组数

第二行开始共T行，每行三个数n m p，意义如上

输出格式：

共T行，每行一个整数表示答案。

输入输出样例

输入样例#1： 复制

2
1 2 5
2 1 5
输出样例#1： 复制

3
3
公式

c(n,m)%p=c(n/p,m/p)*c(n%p,m%p)%p

#include<cstdio>
#include<cctype>
#include<algorithm>
#define ll long long
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if (ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f; 
}
int T;
const int N=1e5+10;
ll inv[N],g[N];int n,m,mod;
inline int c(int n,int m){
    if (n<m) return 0;
    return g[n]*inv[n-m]%mod*inv[m]%mod;
}
int main(){
    freopen("luogu3807.in","r",stdin);
    T=read();inv[1]=1;g[0]=1;inv[0]=1;
    while(T--){
        n=read();m=read();mod=read();static ll ans;ans=1;
        for (int i=1;i<=n+m;++i) g[i]=g[i-1]*i%mod;
        for (int i=2;i<=n+m;++i) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
        for (int i=1;i<=n+m;++i) inv[i]=inv[i-1]*inv[i]%mod;n=n+m;
        for (;n&&m;m/=mod,n/=mod) ans=ans*c(n%mod,m%mod)%mod;printf("%lld\n",ans);
    }
    return 0;
}