首页 > 代码库 > BZOJ 2142 礼物 组合数学+数论

BZOJ 2142 礼物 组合数学+数论

题目大意:给定n个物品,分给m个人,每个人拿到wi个礼物,问方案数mod P P不一定为质数

首先我们把剩下的礼物也分给一个人 答案明显不变 w[++m]=n-w1-w2-...-wm

然后就会很方便地得到公式:

ans=C(n,w1)*C(n-w1,w2)*C(n-w1-w2,w3)*...*C(n-w1-w2-...-w_(m-1),wm) mod P

       =n!/w1!/w2!/.../wm! mod P

然后p不是质数 我们把P分解 令P=∏pi^ai

我们分别处理,可以得到一次同余方程组ans%pi^ai=lefti,用中国剩余定理合并一下即可。

然后对于每个pi^ai,我们进行以下处理:

将分子和分母化为x*pi^y的形式

然后分母的x部分与pi互质,可以求逆元,分子分母的y部分直接相减即可

然后我们处理阶乘

以19为例,将19!化为x*pi^y的形式,其中pi=3,ai=2 则有

19!%9=(1*2*3*4*5*6*7*8*9*10*11*12*13*14*15*16*17*18*19) %9

           =(1*2*4*5*7*8*10*11*13*14*16*17*19)*3^6*(1*2*3*4*5*6) %9

式子的左半部分是不为3的倍数的数,存在长度为p^a的循环节 求出一个循环节 快速幂处理 然后处理剩余部分

右半部分是6!%9 递归处理即可

我这题解写的真是冷静。。。这题还真TM让人冷静不下来啊-0-

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> abcd;
ll P,n,m,a[20],ans;
struct prime_factor{
	ll p,a,p_a,left;
}prime[50];int tot;
void Decomposition(ll x)
{
	ll i;
	for(i=2;i*i<=x;i++)
		if(x%i==0)
		{
			prime[++tot].p=i;
			prime[tot].p_a=1;
			while(x%i==0)
				x/=i,prime[tot].a++,prime[tot].p_a*=i;
		}
	if(x!=1)
		prime[++tot].p=x,prime[tot].a=1,prime[tot].p_a=x;
}
ll Quick_Power(ll x,ll y,ll mo)
{
	ll re=1;
	while(y)
	{
		if(y&1)re*=x,re%=mo;
		x*=x,x%=mo;
		y>>=1;
	}
	return re;
}
abcd Deal(ll x,int pos)
{
	ll i;
	abcd re=make_pair( 1ll , x/prime[pos].p );
	for(i=1;i<prime[pos].p_a;i++)
		if(i%prime[pos].p)
			re.first*=i,re.first%=prime[pos].p_a;
	re.first=Quick_Power(re.first,x/prime[pos].p_a,prime[pos].p_a);
	for(i=x-x%prime[pos].p_a+1;i<=x;i++)
		if(i%prime[pos].p)
			re.first*=i,re.first%=prime[pos].p_a;
	if(re.second)
	{
		abcd temp=Deal(re.second,pos);
		re.first*=temp.first;
		re.first%=prime[pos].p_a;
		re.second+=temp.second;
	}
	return re;
}
abcd EXGCD(ll x,ll y)
{
	if(!y)
		return make_pair(1,0);
	abcd temp=EXGCD(y,x%y);
	return make_pair(temp.second,temp.first-x/y*temp.second);
}
ll Reverse(ll x,ll p)
{
	ll re=EXGCD(x,p).first;
	re=(re%p+p)%p;
	return re;
}
void Chinese_Remainder_Theorem()
{
	int i;
	for(i=1;i<=tot;i++)
	{
		abcd temp=EXGCD(P/prime[i].p_a,prime[i].p_a);
		ll x=temp.first;x=(x%P+P)%P;
		ans+=x*P/prime[i].p_a*prime[i].left;
		ans%=P;
	}
}
int main()
{
	int i,j;
	ll sum=0;
	cin>>P>>n>>m;
	for(i=1;i<=m;i++)
		cin>>a[i],sum+=a[i];
	if(sum>n)
	{
		puts("Impossible");
		return 0;
	}
	if(sum<n)
		a[++m]=n-sum;
	Decomposition(P);
	for(i=1;i<=tot;i++)
	{
		abcd temp=Deal(n,i);
		for(j=1;j<=m;j++)
		{
			abcd _temp=Deal(a[j],i);
			temp.second-=_temp.second;
			temp.first*=Reverse(_temp.first,prime[i].p_a);
			temp.first%=prime[i].p_a;
		}
		temp.first*=Quick_Power(prime[i].p,temp.second,prime[i].p_a);
		prime[i].left=temp.first%prime[i].p_a;
	}
	Chinese_Remainder_Theorem();
	cout<<(ans%P+P)%P<<endl;
}


BZOJ 2142 礼物 组合数学+数论