bzoj3129 [Sdoi2013]方程 容斥+扩展lucas

调了两天,全是低级错误和脑子问题

对于>=的,直接用m减掉

对于<=的,一开始想枚举减多少,但发现对于每个枚举的值是独立的,所以是1e9的

然后由于>=和<=是两个对立的情况,所以直接用容斥,合法的减掉不合法的

但由于我简单容斥做多了,所以就直接用全合法的-全不合法的

然后一定要放弃合法的-不合法的这种想法,应该改成0不合法-1不合法+2不合法-3不合法。。。

然后快速幂还没取模了,exlucas的快速幂还算错了次方


码:

#include<iostream>
#include<cstdio>
using namespace std;
#define ll long long 
ll TREE,x,y,a[9999],P[9999],p[9999],T,cnt,lin,i,m2,n,n2,n1,m,b[99];
void exgcd(ll a,ll b)
{
	if(!b)
	{
		x=1;y=0;return ;
	}
	exgcd(b,a%b);
	ll t=x;
	x=y;
	y=t-(a/b)*y;	
}
ll ksm(ll a,ll b,ll c)
{
	ll ans=1;
	while(b)
	{
	if(b&1)ans=a*ans%c;
	b/=2;
	a=a*a%c;		
	}
	return ans;
}
ll CRT()
{
	int i;
	ll daan=0;
	for(i=1;i<=cnt;i++)
	{
	ll mi=TREE/P[i];
	exgcd(mi,P[i]);
	daan=(daan+x*a[i]%TREE*mi%TREE)%TREE;		
	}	
	if(daan<0)daan+=TREE;
	return daan;
}
struct la
{
	ll gs,ans;
};
la exlucas(ll n,int xh)
{
  	ll ls=n/p[xh],ys=n%P[xh],i;
		la o;
		o.ans=1;
		o.gs=ls;
  if(n==0)return o;
if(n>=P[xh])	for(i=1;i<=P[xh];i++)
  	{
  		 if(i%p[xh]==0)continue;		   
		   o.ans*=i;
  		 o.ans%=P[xh];
	}		
	o.ans=ksm(o.ans,n/P[xh],P[xh]);
for(i=1;i<=ys;i++)
  	{
  		 if(i%p[xh]==0)continue;
  		 o.ans*=i;
		 o.ans%=P[xh];
	}
	la nd=exlucas(ls,xh);
	o.gs+=nd.gs;
	o.ans=o.ans*nd.ans%P[xh];
	return o;
}
ll C(ll n,ll m)
{
	if(n<0||n>m)return 0; 
	int i,ms;
	for(ms=1;ms<=cnt;ms++)
	{ 
	  a[ms]=1;
la fz=exlucas(m,ms);
la fm1=exlucas(n,ms);
la fm2=exlucas(m-n,ms);
fz.gs-=fm1.gs+fm2.gs;
exgcd(fm1.ans,P[ms]);
fz.ans=fz.ans*x%P[ms];
exgcd(fm2.ans,P[ms]);
fz.ans=fz.ans*x%P[ms];
a[ms]=fz.ans*ksm(p[ms],fz.gs,P[ms])%P[ms];
if(a[ms]<0)a[ms]+=P[ms];
	}
	return CRT();
}	ll daan=0;
int jishu=0;
void dfs(ll m2,ll wz)
{
	if(wz==n1+1)
	{
	if(jishu&1)
	{
	daan-=C(n-1,m2-1);	
	daan%=TREE;
	}	else daan+=C(n-1,m2-1),daan%=TREE;
	return ;		
	}
	jishu++;
	dfs(m2-b[wz],wz+1);
	jishu--;
	dfs(m2,wz+1);	
}
int main()
{
	scanf("%lld%lld",&T,&TREE);
	lin=TREE;
	for(i=2;i*i<=lin;i++)
	{
		if(lin%i==0)P[++cnt]=1,p[cnt]=i;
		while(lin%i==0)lin/=i,P[cnt]*=i;
	}
	if(lin!=1)P[++cnt]=p[cnt]=lin;
	while(T--)
	{
	daan=0;
		scanf("%lld%lld%lld%lld",&n,&n1,&n2,&m);
	m2=m;
		for(i=1;i<=n1;i++)
		scanf("%lld",&b[i]);
		for(i=1;i<=n2;i++)
		scanf("%lld",&x),m-=(x-1),m2-=(x-1);
         dfs(m,1);
         printf("%lld\n",(daan+TREE)%TREE);
	}	
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值