Lucas定理板子 洛谷P3807

数论Lucas定理是用来求 C(m,n)%p的值,【p是素数】

【辣鸡杭电针对我,在洛谷上能A在杭电上就要T】
HDU3037还涉及到隔板法,详情见此
洛谷直接给了公式。
1、O(n)预处理逆元

#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
const int N=1e5+5;
/*ll fac[N],facinv[N],inv[N];
void getinv(int n,int p)
{
    inv[1]=1;
    facinv[0]=1,facinv[1]=1;
    fac[0]=1,fac[1]=1;
    for(int i=2;i<=n;i++){
        fac[i]=fac[i-1]*i%p;
        inv[i]=(p-p/i)*inv[p%i]%p;
        facinv[i]=facinv[i-1]*inv[i]%p;
    }
}*/
void exgcd(int a,int b,int &x,int &y)
{
    if(!b){
        x=1,y=0;
        return;
    } 
    int xx,yy;
    exgcd(b,a%b,xx,yy);
    x=yy;
    y=xx-a/b*yy;
}
ll inv(int a,int p)
{
    int x,y;
    exgcd(a,p,x,y);
    return (x%p+p)%p;
}
ll comb(int n,int m,int p)
{
    ll ans=1;
    for(int i=m+1;i<=n;i++) ans=ans*i%p;
    for(int i=1;i<=n-m;i++)
        ans=ans*inv(i,p)%p;
    return ans;
}
ll Lucas(int n,int m,int p)
{
    if(n<m) return 0;
    if(n<p&&m<p)  return comb(n,m,p);
    return Lucas(n%p,m%p,p)*Lucas(n/p,m/p,p)%p;
} 
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        int n,m,p;
        scanf("%d%d%d",&n,&m,&p);
    //  getinv(p-1,p);
        printf("%lld\n",Lucas(n+m,m,p));
    }
    return 0;
}

2、在线求逆元

#include<cstdio>
#include<cstring>
using namespace std;
#define ll long long
const int N=1e5+5;
/*ll fac[N],facinv[N],inv[N];
void getinv(int n,int p)
{
    inv[1]=1;
    facinv[0]=1,facinv[1]=1;
    fac[0]=1,fac[1]=1;
    for(int i=2;i<=n;i++){
        fac[i]=fac[i-1]*i%p;
        inv[i]=(p-p/i)*inv[p%i]%p;
        facinv[i]=facinv[i-1]*inv[i]%p;
    }
}*/
void exgcd(int a,int b,int &x,int &y)
{
    if(!b){
        x=1,y=0;
        return;
    } 
    int xx,yy;
    exgcd(b,a%b,xx,yy);
    x=yy;
    y=xx-a/b*yy;
}
ll inv(int a,int p)
{
    int x,y;
    exgcd(a,p,x,y);
    return (x%p+p)%p;
}
ll comb(int n,int m,int p)
{
    ll ans=1;
    for(int i=m+1;i<=n;i++) ans=ans*i%p;
    for(int i=1;i<=n-m;i++)
        ans=ans*inv(i,p)%p;
    return ans;
}
ll Lucas(int n,int m,int p)
{
    if(n<m) return 0;
    if(n<p&&m<p)  return comb(n,m,p);
    return Lucas(n%p,m%p,p)*Lucas(n/p,m/p,p)%p;
} 
int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        int n,m,p;
        scanf("%d%d%d",&n,&m,&p);
        printf("%lld\n",Lucas(n+m,m,p));
    }
    return 0;
}

两者时间复杂度的比较:
上面的是O(n)求逆元的
下面是在线求逆元的【更快】
这里写图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值