题意:
给你一个方程:
x1+x2+x3+……xn=m
的整数解个数。
其中前
n1
个数要
<=ai
<script type="math/tex" id="MathJax-Element-123"><=a_i</script>后
n2
个数
>=ai+n1
题解:
要大于某个数,就直接m减去
ai−1
就好了。
然后用隔板法,在容斥一下,就可以了。
至于为什么可以扩展lucas,不用管的了反正就可以了
code:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#define LL long long
using namespace std;
LL MOD;
LL p_i[25],p_k[25],num=0;
void pre()
{
LL x=MOD;
for(LL i=2;i*i<=x;i++)
if(x%i==0)
{
LL k=1;
while(x%i==0) x/=i,k*=i;
p_i[++num]=i;p_k[num]=k;
}
if(x!=1) p_i[++num]=x,p_k[num]=x;
}
LL pow(LL a,LL b,LL mod)
{
LL ans=1;
while(b)
{
if(b&1) ans=ans*a%mod;
a=a*a%mod;b>>=1;
}
return ans;
}
LL inv(LL a,LL pi,LL pk)
{
if(!a) return 1;
return pow(a,pk-(pk/pi)-1,pk);
}
LL mul(LL n,LL pi,LL pk)
{
if(!n) return 1;
LL ans=1;
for(LL i=2;i<=pk;i++)
if(i%pi) ans=ans*i%pk;
ans=pow(ans,n/pk,pk);
for(LL i=2;i<=n%pk;i++)
if(i%pi) ans=ans*i%pk;
return ans*mul(n/pi,pi,pk)%pk;
}
LL C(LL n,LL m,LL mod,LL pi,LL pk)
{
if(m>n) return 0;
LL a=mul(n,pi,pk),b=mul(m,pi,pk),c=mul(n-m,pi,pk);
LL ans,k=0;
for(LL i=n;i;i/=pi) k+=i/pi;
for(LL i=m;i;i/=pi) k-=i/pi;
for(LL i=n-m;i;i/=pi) k-=i/pi;
ans=a*inv(b,pi,pk)%pk*inv(c,pi,pk)%pk*pow(pi,k,pk)%pk;
return ans*(mod/pk)%mod*inv(mod/pk,pi,pk)%mod;
}
LL lucas(LL n,LL m)
{
if(m>n) return 0;
LL ans=0;
for(LL i=1;i<=num;i++) ans=(ans+C(n,m,MOD,p_i[i],p_k[i]))%MOD;
return ans;
}
LL n,m,n1,n2,a[15];
LL ans;
void dfs(LL x,LL k,LL num)
{
if(k<n) return;
if(x>n1)
{
if(num&1) ans-=lucas(k-1,n-1);
else ans+=lucas(k-1,n-1);
ans%=MOD;
return;
}
dfs(x+1,k,num);
dfs(x+1,k-a[x],num+1);
}
int main()
{
LL T;scanf("%lld %lld",&T,&MOD);
pre();
while(T--)
{
scanf("%lld %lld %lld %lld",&n,&n1,&n2,&m);
for(LL i=1;i<=n1;i++) scanf("%lld",&a[i]);
for(LL i=1;i<=n2;i++)
{
LL x;scanf("%lld",&x);
m-=(x-1);
}
ans=0;
dfs(1,m,0);
printf("%lld\n",(ans+MOD)%MOD);
}
}