ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

[模板]扩展卢卡斯定理

2022-07-23 12:33:17  阅读:151  来源: 互联网

标签:md ch tot 定理 long WR 卢卡斯 include 模板


#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#define WR WinterRain
using namespace std;
const long long WR=10010;
long long n0,m0,w[WR],mod;
long long a[WR],b[WR];
long long res=1;
long long read(){
    long long s=0,w=1;
    char ch=getchar();
    while(ch>'9'||ch<'0'){
        if(ch=='-') w=-1;
        ch=getchar();
    }
    while(ch<='9'&&ch>='0'){
        s=(s<<3)+(s<<1)+ch-48;
        ch=getchar();
    }
    return s*w;
}
long long quick_pow(long long a,long long b,long long md){
    long long base=a,ans=1;
    while(b>0){
        if(b&1) ans=ans*base%md;
        base=base*base%md;
        b>>=1;
    }
    return ans%md;
}
long long ex_gcd(long long &x,long long &y,long long a,long long b){
    if(b==0){
        x=1,y=0;
        return a;
    }
    long long d=ex_gcd(x,y,b,a%b);
    long long tmp=x;
    x=y,y=tmp-a/b*y;
    return d;
}
long long CRT(long long n,long long *a,long long *b){
    long long md=1,ans=0;
    for(long long i=1;i<=n;i++) md=md*b[i];
    for(long long i=1;i<=n;i++){
        long long x,y;
        ex_gcd(x,y,md/b[i],b[i]);
        ans=(ans+md/b[i]*a[i]*x)%md;
    }
    return (ans+md)%md;
}
//=============EX_LUCAS板子===============
long long calc(long long n,long long p,long long md){
    //printf("calculating %lld %lld %lld\n",n,p,md);
    if(!n) return 1;
    long long ans=1;
    for(long long i=1;i<=md;i++)
        if(i%p!=0) ans=ans*i%md;
    ans=quick_pow(ans,n/md,md);
    for(long long i=1;i<=n%md;i++)
        if(i%p!=0) ans=ans*i%md;
    ans=ans*calc(n/p,p,md)%md;
    return ans;
}
long long ex_Lucas(long long n,long long m,long long p,long long md){
    //printf("EX_Lucasing %lld %lld %lld %lld\n",n,m,p,md);
    long long cnt=0;
    for(long long i=n;i;i/=p) cnt+=i/p;
    for(long long i=m;i;i/=p) cnt-=i/p;
    for(long long i=n-m;i;i/=p) cnt-=i/p;
    return quick_pow(p,cnt,md)*calc(n,p,md)%md
           *quick_pow(calc(m,p,md),md/p*(p-1)-1,md)%md
           *quick_pow(calc(n-m,p,md),md/p*(p-1)-1,md)%md;
}
long long comb(long long n,long long m,long long md){
    memset(a,0,sizeof(a));
    memset(b,0,sizeof(b));
    //printf("combing %lld %lld %lld\n",n,m,md);
    long long tot=0;
    for(long long i=2;i*i<=md;i++){
        if(md%i==0){
            b[++tot]=1;
            while(md%i==0) md/=i,b[tot]*=i;
            a[tot]=ex_Lucas(n,m,i,b[tot]);
            //printf("%lld calculate completed\n",i);
        }
    }
    if(md>1) b[++tot]=md,a[tot]=ex_Lucas(n,m,md,b[tot]);
    //for(long long i=1;i<=tot;i++){
    //    printf("%lld %lld\n",a[i],b[i]);
    //}
    return CRT(tot,a,b);
}
int main(){
    n0=read(),m0=read(),mod=read();
    printf("%lld",comb(n0,m0,mod));
    return 0;
}

 

标签:md,ch,tot,定理,long,WR,卢卡斯,include,模板
来源: https://www.cnblogs.com/WintersRain/p/16218060.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有