【模板】卢卡斯定理/Lucas 定理

最新推荐文章于 2024-08-26 21:46:59 发布

green qwq

最新推荐文章于 2024-08-26 21:46:59 发布

阅读量828

点赞数 8

文章标签：算法

本文链接：https://blog.csdn.net/m0_60738889/article/details/139888538

版权

【模板】卢卡斯定理/Lucas 定理

题目背景

这是一道模板题。

题目描述

给定整数 $n, m, p$ 的值，求出 $C_{n + m}^n \bmod p$ 的值。

输入数据保证 $p$ 为质数。

注: $C$ 表示组合数。

输入格式

本题有多组数据。

第一行一个整数 $T$ ，表示数据组数。

对于每组数据:

一行，三个整数 $n, m, p$ 。

输出格式

对于每组数据，输出一行，一个整数，表示所求的值。

样例 #1

样例输入 #1

2
1 2 5
2 1 5

样例输出 #1

3
3

提示

对于 $100\%$ 的数据， $\leq n, m, p \leq 10^5$ ， $\leq T \leq 10$ 。

模板

//lucas定理

#include<iostream>
#include<algorithm>
#include<cstring>

#define int long long

using namespace std;

const int N = 1e5+10;

int infac[N],fac[N];
int T;
int n,m,p;

int qmi(int a,int b,int p){
    int res=1;
    
    while(b){
        if(b&1)res=res*a%p;
        a=a*a%p;
        b>>=1;
    }
    return res;
}

void init(){
    fac[0]=infac[0]=1;
    
    for(int i=1;i<N-10;i++){
        fac[i]=fac[i-1]*i%p;
        infac[i]=infac[i-1]*qmi(i,p-2,p)%p;
    }
}

int C(int a,int b,int p){
    int ans=0;
    if(b>a)return 0;
    
    return fac[a]%p*infac[b]%p*infac[a-b]%p;
    
}

int lucas(int a,int b,int p){
    if(a<p&&b<p)return C(a,b,p);
    return C(a%p,b%p,p)*lucas(a/p,b/p,p)%p;
}

signed main(){
    cin>>T;
    
    while(T--){
        cin>>n>>m>>p;
        init();
        
        cout<<lucas(n+m,m,p)<<endl;
    }
    return 0;
}