如果没有限制,答案直接用隔板法C(m-1,n-1)
对于>=x的限制,我们直接在对应位置先放上x-1即可,即m=m-(x-1)
对于<=x的限制,由于限制很小我们可以利用容斥原理将它转化为上面的>=x的限制
即减去1个不满足的 加上2个不满足的 减去3个不满足的 ……
之后就是组合数的计算,对于一个非常大的模数,我们可以将它唯一分解,之后CRT还原即可
但是我们有可能不存在逆元,数据范围不允许我们递推计算组合数
我们知道没有逆元当且仅当(a,p)不互素,我们可以将阶乘分成两部分:互素和不互素
互素的部分具有循环节,我们暴力计算循环节之后快速幂即可
不互素的部分我们可以提出他们的gcd出来并记录指数,剩余部分又变成了计算阶乘,递归即可
#include<cstdio> #include<cstring> #include<iostream> #include<cstdlib> #include<algorithm> using namespace std; const int maxn=110; typedef long long LL; int T,tot=0; LL p,n,n1,n2,m; LL mod[maxn],prime[maxn],a[maxn]; LL Num[maxn]; LL x,y,d; LL ans; void Get_mod(LL x){ for(int i=2;i*i<=x;++i){ if(x%i==0){ prime[++tot]=i;mod[tot]=1; while(x%i==0){x/=i;mod[tot]*=i;} } } if(x>1)prime[++tot]=x,mod[tot]=x; } void ex_gcd(LL a,LL b,LL &x,LL &y,LL &d){ if(b==0){x=1;y=0;d=a;} else{ex_gcd(b,a%b,y,x,d);y-=(a/b)*x;} } LL pow_mod(LL v,LL p,LL mod){ LL tmp=1; while(p){ if(p&1)tmp=tmp*v%mod; v=v*v%mod;p>>=1; }return tmp; } LL inv(LL a,LL b){ ex_gcd(a,b,x,y,d); return (x%b+b)%b; } LL CRT(){ LL ans=0; for(int i=1;i<=tot;++i){ LL m=p/mod[i]; ex_gcd(mod[i],m,d,y,d); ans=(ans+y*m*a[i])%p; }return (ans+p)%p; } pair<LL,LL> fac(int k,LL n){ if(n==0)return make_pair(0,1); int x=n/prime[k],y=n/mod[k]; LL ans=1; if(y){ for(int i=2;i<mod[k];++i)if(i%prime[k]!=0)ans=(ans*1LL*i)%mod[k]; ans=pow_mod(ans,y,mod[k]); } for(int i=y*mod[k]+1;i<=n;++i)if(i%prime[k]!=0)ans=(ans*1LL*i)%mod[k]; pair<LL,LL> tmp=fac(k,x); return make_pair(x+tmp.first,ans*tmp.second%mod[k]); } LL cal(int k,LL n,LL m){ if(n<m)return 0; pair<LL,LL> a=fac(k,n),b=fac(k,m),c=fac(k,n-m); return pow_mod(prime[k],a.first-b.first-c.first,mod[k])*a.second%mod[k] *inv(b.second,mod[k])%mod[k]*inv(c.second,mod[k])%mod[k]; } LL C(LL n,LL m){ for(int i=1;i<=tot;++i)a[i]=cal(i,n,m); return CRT(); } void DFS(int pos,int now,LL sum){ if(pos>n1){ if(now)ans-=C(m-1-sum,n-1); else ans+=C(m-1-sum,n-1); ans=(ans+p)%p; return; } DFS(pos+1,now,sum); DFS(pos+1,now^1,sum+Num[pos]); } int main(){ scanf("%d%lld",&T,&p); Get_mod(p); while(T--){ scanf("%lld%lld%lld%lld",&n,&n1,&n2,&m); for(int i=1;i<=n1;++i)scanf("%lld",&Num[i]); for(int i=1;i<=n2;++i){scanf("%lld",&x);m=m-x+1;} ans=0; DFS(1,0,0); printf("%lld\n",(ans+p)%p); }return 0; }
时间: 2024-10-13 23:28:44