[BZOJ]2142: 礼物 扩展lucas

Description
一年一度的圣诞节快要来到了。每年的圣诞节小E都会收到许多礼物，当然他也会送出许多礼物。不同的人物在小E心目中的重要性不同，在小E心中分量越重的人，收到的礼物会越多。小E从商店中购买了n件礼物，打算送给m个人，其中送给第i个人礼物数量为wi。请你帮忙计算出送礼物的方案数（两个方案被认为是不同的，当且仅当存在某个人在这两种方案中收到的礼物不同）。由于方案数可能会很大，你只需要输出模P后的结果。

题解：

会扩展lucas的话就很简单了，看zyf2000的博客很快就学会了，强烈推荐。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
LL read()
{
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
LL Pow(LL x,LL y,LL mod)
{
    if(y==0)return 1;
    if(y==1)return x;
    LL t=Pow(x,y>>1,mod),ans=t*t%mod;
    if(y&1)ans=ans*x%mod;
    return ans;
}
LL exgcd(LL a,LL b,LL &x,LL &y)
{
    if(!a)
    {
        x=0,y=1;
        return b;
    }
    LL tx,ty,d=exgcd(b%a,a,tx,ty);
    y=tx;
    x=ty-(b/a)*tx;
    return d;
}
LL inv(LL a,LL mod)
{
    if(!a)return 1;
    LL x,y,d=exgcd(a,mod,x,y);
    return (x%mod+mod)%mod;
}
LL fac(LL n,LL p,LL pk)
{
    if(!n)return 1;
    LL re=1;
    for(LL i=2;i<=pk;i++)
    if(i%p)re=re*i%pk;
    re=Pow(re,n/pk,pk);
    for(LL i=2;i<=n%pk;i++)
    if(i%p)re=re*i%pk;
    return re*fac(n/p,p,pk)%pk;
}
LL C(LL n,LL m,LL p,LL pk)
{
    if(n<m)return 0;
    LL a=fac(n,p,pk),b=fac(m,p,pk),c=fac(n-m,p,pk),cnt=0;
    for(LL i=n;i;i/=p)cnt+=(i/p);
    for(LL i=m;i;i/=p)cnt-=(i/p);
    for(LL i=n-m;i;i/=p)cnt-=(i/p);
    return a*inv(b,pk)%pk*inv(c,pk)%pk*Pow(p,cnt,pk)%pk;
}
LL p,n,m,w[7];
LL lucas(LL n,LL m,LL mod)
{
    LL M=mod,re=0;
    for(LL i=2;i*i<=mod;i++)
    if(mod%i==0)
    {
        LL t=1;
        while(mod%i==0)mod/=i,t*=i;
        re=(re+C(n,m,i,t)*inv(M/t,t)%M*(M/t)%M)%M;
    }
    if(mod>1)re=(re+C(n,m,mod,mod)*inv(M/mod,mod)%M*(M/mod)%M)%M;
    return re;
}
int main()
{
    LL ans=1,ss=0;
    p=read(),n=read(),m=read();
    for(LL i=1;i<=m;i++)w[i]=read(),ss+=w[i];
    if(n<ss)return puts("Impossible"),0;
    for(LL i=1;i<=m;i++)
    {
        ans=ans*lucas(n,w[i],p)%p;
        n-=w[i];
    }
    printf("%lld",ans);
}