首页 > 代码库 > HDU 5446 Unknown Treasure(Lucas定理+CRT)

HDU 5446 Unknown Treasure(Lucas定理+CRT)

 

【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=5446

 

【题目大意】

  给出一个合数M的每一个质因子,同时给出n,m,求C(n,m)%M。

 

【题解】

  首先我们可以用Lucas定理求出对答案对每个质因子的模,然后我们发现只要求解这个同余方程组就可以得到答案,所以我们用中国剩余定理解决剩下的问题。

 

【代码】

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;typedef long long LL;const int N=100005;LL f[N],rf[N];LL mul(LL x,LL y,LL P){return (x*y-(LL)(x/(long double)P*y+1e-3)*P+P)%P;}LL pow(LL a,LL b,LL P){LL t=1;for(;b;b>>=1,a=mul(a,a,P))if(b&1)t=mul(t,a,P);return t;}void init(int n){    f[0]=1;for(int i=1;i<n;i++)f[i]=f[i-1]*i%n;    rf[n-1]=pow(f[n-1],n-2,n);    for(int i=n-1;i;i--)rf[i-1]=rf[i]*i%n;}LL C(int n,int m,int mod){    if(m>n||m<0||n<0)return    0;    return f[n]*rf[m]%mod*rf[n-m]%mod;}LL lucas(LL n,LL m,LL P){    if(n<m)return 0;    if(!m||n==m)return 1;    return C(n%P,m%P,P)*lucas(n/P,m/P,P)%P;}LL exgcd(LL a,LL b,LL& x,LL& y){    if(b==0){x=1;y=0;return a;}    LL d=exgcd(b,a%b,y,x);    y-=x*(a/b);    return d;}LL CRT(LL* a,LL* b,int n){    LL P=1,d,y,x=0;    for(int i=0;i<n;i++)P*=a[i];    for(int i=0;i<n;i++){        LL w=P/a[i];        exgcd(a[i],w,d,y); y=(y%P+P)%P;         x=(x+mul(mul(y,w,P),b[i],P));    }return (x+P)%P;}int T,k;LL n,m,a[15],p[15];int main(){    scanf("%d",&T);    while(T--){        scanf("%lld%lld%d",&n,&m,&k);        for(int i=0;i<k;i++){            scanf("%lld",&p[i]);            init(p[i]);            a[i]=lucas(n,m,p[i]);        }printf("%lld\n",CRT(p,a,k));    }return 0;}

 

  

 

HDU 5446 Unknown Treasure(Lucas定理+CRT)