●BZOJ 3129 [Sdoi2013]方程

时间:2022-06-09 14:53:32

题链:

http://www.lydsy.com/JudgeOnline/problem.php?id=3129

题解:

容斥,扩展Lucas,中国剩余定理

先看看不管限制,只需要每个位置都是正整数时的方案数的求法。
假设有 N 个未知数,加起来的和为 M。
转化一下问题变为:"小球分配"
有 M 个相同的小球,放在 N 个盒子里,且每个盒子至少有一个的方案数。
那么方案数为 ${C}_{M-1}^{N-1}$
怎么理解这个式子呢?"插隔板法"。
使 M个小球放在一排,考虑可以在相邻小球的空隙间(共 M-1个空隙)插入一个隔板,总共插入 N-1个隔板。
把相邻的两个隔板(把首尾也看作另外2个隔板)中间的区域看做一个个的盒子,区域内的小球即为该盒子里的小球。
则任意一种插隔板的方法都对应一种把小球放入盒子的方法。
所以方案数为
${C}_{M-1}^{N-1}$

对于题目给的两种限制,不要被吓到了。
其实大于等于(>=W)这一个限制很好处理,直接把 M-=(W-1),即事先给这些位置分配 W-1个小球。
因为接下来按照上面的 "小球分配" 方式,这些限制一定可以满足。
然后对于小于等于(<=W) 就需要用到容斥了。
定义 f[S] 表示至少 S 集合里面的这些"小于等于"限制都不能满足,那么方案数的求法就是:
先强制给这些盒子分配 Wi 个小球,
使得接下来按照上面的 "小球分配" 方式求出方案数,这些盒子一定会超过限制。
所以按照容斥的做法:
答案 ANS = 至少 0 个盒子超出限制的方案数
                - 至少 1 个盒子超出限制的方案数
                +至少 2 个盒子超出限制的方案数
                -...+...

但是求那个方案数 C(N,M) 就比较麻烦了,因为 N,M很大,且模数不保证为质数,
所以用到 扩展Lucas(+国剩) 来求。

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#define _ % P
#define filein(x) freopen(#x".in","r",stdin);
#define fileout(x) freopen(#x".out","w",stdout);
using namespace std;
int h[300],k[300],d[300],pi[10],ppi[10],fc[10][100500],pnt;
int T,N,N1,N2,M,ANS;
void init(){
ANS=0;
memset(h,0,sizeof(h));
memset(k,0,sizeof(k));
memset(d,0,sizeof(d));
}
void pre_prime(int P){
static bool np[100500];
static int prime[100500],cnt;
for(int i=2;i<=100000;i++){
if(!np[i]){
prime[++cnt]=i;
if(P%i==0){
pi[++pnt]=i; ppi[pnt]=1;
while(P%i==0) ppi[pnt]*=i,P/=i;
}
}
for(int j=1;j<=cnt&&i<=100000/prime[j];j++){
np[i*prime[j]]=1;
if(i%prime[j]==0) break;
}
}
for(int i=1;i<=pnt;i++){
fc[i][0]=1;
for(int j=1;j<=100000;j++)
fc[i][j]=(1ll*fc[i][j-1]*(j%pi[i]?j:1))%ppi[i];
}
}
int pow(int a,int b,int P){
int ret=1; a=(a)_;
while(b){
if(b&1) ret=(1ll*ret*a)_;
a=(1ll*a*a)_; b>>=1;
}
return ret;
}
void exEuclidean(int a,int b,int &g,long long &x,long long &y){
if(!b) g=a,x=1,y=0;
else exEuclidean(b,a%b,g,y,x),y-=(a/b)*x;
}
int inv(int a,int P){
int g; long long x,y;
exEuclidean(a,P,g,x,y);
x=((1ll*x)_+P)_;
return (int)x;
}
int fac(int n,int I,int Pi,int P){
if(!n) return 1; int ret=1,r=n%P;
//for(int i=2;i<=P;i++) if(i%Pi) ret=(1ll*ret*i)_;
ret=fc[I][P];
ret=pow(ret,n/P,P);
//for(int i=2;i<=r;i++) if(i%Pi) ret=(1ll*ret*i)_;
ret=(1ll*ret*fc[I][r])_;
return (1ll*ret*fac(n/Pi,I,Pi,P))_;
}
int C(int n,int m,int I,int Pi,int P){
int x,y,z,c=0;
x=fac(n,I,Pi,P); y=fac(m,I,Pi,P); z=fac(n-m,I,Pi,P);
for(int i=n;i;i/=Pi) c+=i/Pi;
for(int i=m;i;i/=Pi) c-=i/Pi;
for(int i=n-m;i;i/=Pi) c-=i/Pi;
return ((((1ll*x*inv(y,P))_)*inv(z,P))_*pow(Pi,c,P))_;
}
int exLucas(int n,int m,int P){
int ret=0,tmp; if(n<m) return 0;
for(int i=1;i<=pnt;i++){
tmp=C(n,m,i,pi[i],ppi[i]);
tmp=((1ll*tmp*(P/ppi[i]))_*inv(P/ppi[i],ppi[i]))_;
ret=(1ll*ret+tmp)_;
}
return ret;
}
int main()
{
int P;
scanf("%d%d",&T,&P);
pre_prime(P);
while(T--){
init();
scanf("%d%d%d%d",&N,&N1,&N2,&M);
for(int i=1;i<=N1;i++) scanf("%d",&d[1<<(i-1)]);
for(int i=1,x;i<=N2;i++) scanf("%d",&x),M-=(x-1);
for(int s=1,p;p=s&-s,s<(1<<N1);s++)
h[s]=h[s^p]+1,k[s]=k[s^p]+d[p];
for(int s=0,m,tmp;s<(1<<N1);s++){
m=M-k[s];
tmp=exLucas(m-1,N-1,P);
if(h[s]&1) tmp=(-1ll*tmp+P)_;
ANS=((1ll*ANS+tmp)_+P)_;
}
printf("%d\n",ANS);
}
return 0;
}