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);
}